From 24f11b6bea3e3a7988eff31bba61946a0e52fbbe Mon Sep 17 00:00:00 2001 From: osanseviero Date: Mon, 25 Dec 2023 10:26:26 +0100 Subject: [PATCH] Quick test for windows --- metrics/confusion_matrix/confusion_matrix.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py index 1d0c7797..37e0f78f 100644 --- a/metrics/confusion_matrix/confusion_matrix.py +++ b/metrics/confusion_matrix/confusion_matrix.py @@ -40,7 +40,9 @@ >>> confusion_matrix_metric = evaluate.load("confusion_matrix") >>> results = confusion_matrix_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0]) >>> print(results) # doctest: +NORMALIZE_WHITESPACE - {'confusion_matrix': array([[1, 0, 1], [0, 2, 0], [1, 1, 0]])} + {'confusion_matrix': array([[1, 0, 1], + [0, 2, 0], + [1, 1, 0]], dtype=int64)} """