Update confusion_matrix.py

huggingface · Dec 25, 2023 · 3b2a229 · 3b2a229
1 parent 7b7041e
commit 3b2a229
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py
@@ -42,7 +42,7 @@
         >>> print(results)  # doctest: +NORMALIZE_WHITESPACE
         {'confusion_matrix': array([[1, 0, 1],
                             [0, 2, 0],
-                            [1, 1, 0]], dtype=int64)}
+                            [1, 1, 0]])}
 """