Skip to content

GitLab

Unverified Commit b5ba758b authored by gijswijnholds's avatar gijswijnholds Committed by GitHub
Browse files

Import accuracy_score (#6480)

parent 9a8c168f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/training.rst
View file @ b5ba758b
...@@ -282,7 +282,7 @@ your own ``compute_metrics`` function and pass it to the trainer. ...@@ -282,7 +282,7 @@ your own ``compute_metrics`` function and pass it to the trainer.
.. code-block:: python .. 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): def compute_metrics(pred):
labels = pred.label_ids labels = pred.label_ids
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment