Import accuracy_score (#6480)

b5ba758b · gijswijnholds · GitHub · 9a8c168f · b5ba758b
Unverified Commit b5ba758b authored Aug 14, 2020 by gijswijnholds Committed by GitHub Aug 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docs/source/training.rst docs/source/training.rst +1 -1

No files found.
--- a/docs/source/training.rst
+++ b/docs/source/training.rst
@@ -282,7 +282,7 @@ your own ``compute_metrics`` function and pass it to the trainer.

 .. code-block:: python

-    from sklearn.metrics import precision_recall_fscore_support
+    from sklearn.metrics import accuracy_score, precision_recall_fscore_support

    def compute_metrics(pred):
        labels = pred.label_ids