diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py index b118e067..5ea33038 100644 --- a/metrics/confusion_matrix/confusion_matrix.py +++ b/metrics/confusion_matrix/confusion_matrix.py @@ -39,7 +39,7 @@ Example 1-A simple example >>> 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: +ELLIPSIS +NORMALIZE_WHITESPACE + >>> print(results) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE {'confusion_matrix': array([[1, 0, 1], [0, 2, 0], [1, 1, 0]][...])} """