From b5ba758ba983af2b0486dda02ae68109846afdb7 Mon Sep 17 00:00:00 2001 From: gijswijnholds Date: Fri, 14 Aug 2020 14:16:16 +0200 Subject: [PATCH] Import accuracy_score (#6480) --- docs/source/training.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/training.rst b/docs/source/training.rst index 7fe1498472..5d0cbe982b 100644 --- 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