🚧 Work in progress.

BrikerMan · May 10, 2020 · 6f94d06 · 6f94d06
1 parent e67c317
commit 6f94d06
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/kashgari/metrics/multi_label_classification.py b/kashgari/metrics/multi_label_classification.py
@@ -81,8 +81,4 @@ def multi_label_classification_report(y_true: MultiLabelClassificationLabelVar,
 
 
 if __name__ == "__main__":
-    from kashgari.utils import MultiLabelBinarizer
-
-    m = MultiLabelBinarizer({'s': 0, 'c': 1})
-    print(multi_label_classification_report([['s', 'c']], [['s']],
-                                            binarizer=m))
+    pass
diff --git a/kashgari/tasks/classification/abc_model.py b/kashgari/tasks/classification/abc_model.py
@@ -266,9 +266,9 @@ def evaluate(self,  # type: ignore[override]
                 logger.debug('y_pred : {}'.format(y_pred[index]))
 
         if self.multi_label:
-            report = multi_label_classification_report(y_data,
-                                                       y_pred,
-                                                       binarizer=self.label_processor.multi_label_binarizer)
+            report = multi_label_classification_report(y_data,  # type: ignore
+                                                       y_pred,  # type: ignore
+                                                       binarizer=self.label_processor.multi_label_binarizer)  # type: ignore
 
         else:
             original_report = metrics.classification_report(y_data,