From 6f94d06aff01a16f2cac8f3ce4f1bd1b530ed1e2 Mon Sep 17 00:00:00 2001 From: BrikerMan Date: Sun, 10 May 2020 21:59:34 +0800 Subject: [PATCH] :construction: Work in progress. --- kashgari/metrics/multi_label_classification.py | 6 +----- kashgari/tasks/classification/abc_model.py | 6 +++--- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/kashgari/metrics/multi_label_classification.py b/kashgari/metrics/multi_label_classification.py index 9c75f04d..3b91992b 100644 --- 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 index 9edcc66b..7c25a8f6 100644 --- 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,