Achilleas Pasias commited on
Commit
8eb6170
1 Parent(s): d7b729a

Return as numpy to support ravel functionality

Browse files
Files changed (1) hide show
  1. confusion_matrix.py +1 -1
confusion_matrix.py CHANGED
@@ -145,5 +145,5 @@ class ConfusionMatrix(evaluate.Metric):
145
  labels=labels,
146
  sample_weight=sample_weight,
147
  normalize=normalize,
148
- ).tolist()
149
  }
 
145
  labels=labels,
146
  sample_weight=sample_weight,
147
  normalize=normalize,
148
+ )
149
  }