TST Include multilabel_confusion_matrix in test_common.py

raghavrv · raghavrv · commit 72fc7b0f230e · 2015-01-20T06:09:41.000+05:30
diff --git a/sklearn/metrics/tests/test_common.py b/sklearn/metrics/tests/test_common.py
@@ -26,6 +26,7 @@
 from sklearn.metrics import accuracy_score
 from sklearn.metrics import average_precision_score
 from sklearn.metrics import confusion_matrix
+from sklearn.metrics import multilabel_confusion_matrix
 from sklearn.metrics import coverage_error
 from sklearn.metrics import explained_variance_score
 from sklearn.metrics import f1_score
@@ -98,6 +99,7 @@
     "accuracy_score": accuracy_score,
     "unnormalized_accuracy_score": partial(accuracy_score, normalize=False),
     "confusion_matrix": confusion_matrix,
+    "multilabel_confusion_matrix": multilabel_confusion_matrix,
     "hamming_loss": hamming_loss,
 
     "jaccard_similarity_score": jaccard_similarity_score,
@@ -185,6 +187,9 @@
 METRIC_UNDEFINED_MULTICLASS = [
     "samples_f0.5_score", "samples_f1_score", "samples_f2_score",
     "samples_precision_score", "samples_recall_score",
+    # FIXME to be moved to METRIC_UNDEFINED_BINARY after merge of #4115
+    # this is needed to run some basic tests on multilabel_confusion_matrix.
+    "multilabel_confusion_matrix",
 
     # Those metrics don't support multiclass outputs
     "average_precision_score", "weighted_average_precision_score",
@@ -331,6 +336,7 @@
 # No Sample weight support
 METRICS_WITHOUT_SAMPLE_WEIGHT = [
     "confusion_matrix",
+    "multilabel_confusion_matrix",
     "hamming_loss",
     "matthews_corrcoef_score",
     "median_absolute_error",
