暴露更多类

fastNLP/__init__.py

@@ -52,6 +52,7 @@ __all__ = [
     "SpanFPreRecMetric",
     "CMRC2018Metric",
     "ClassifyFPreRecMetric",
+    "ConfusionMatrixMetric",
     
     "Optimizer",
     "SGD",

fastNLP/core/__init__.py

@@ -68,6 +68,7 @@ __all__ = [
     "SpanFPreRecMetric",
     "CMRC2018Metric",
     "ClassifyFPreRecMetric",
+    "ConfusionMatrixMetric",
 
     "Optimizer",
     "SGD",
@@ -90,7 +91,8 @@ from .dataset import DataSet
 from .field import FieldArray, Padder, AutoPadder, EngChar2DPadder
 from .instance import Instance
 from .losses import LossFunc, CrossEntropyLoss, L1Loss, BCELoss, NLLLoss, LossInForward, CMRC2018Loss, LossBase
-from .metrics import AccuracyMetric, SpanFPreRecMetric, CMRC2018Metric, ClassifyFPreRecMetric, MetricBase
+from .metrics import AccuracyMetric, SpanFPreRecMetric, CMRC2018Metric, ClassifyFPreRecMetric, MetricBase,\
+    ConfusionMatrixMetric
 from .optimizer import Optimizer, SGD, Adam, AdamW
 from .sampler import SequentialSampler, BucketSampler, RandomSampler, Sampler
 from .tester import Tester