From 9b77ad812c33c72497491a0046fd207cb76c435e Mon Sep 17 00:00:00 2001 From: yh_cc Date: Sat, 28 Mar 2020 21:41:53 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9A=B4=E9=9C=B2=E6=9B=B4=E5=A4=9A=E7=B1=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/__init__.py | 1 + fastNLP/core/__init__.py | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/fastNLP/__init__.py b/fastNLP/__init__.py index 85b9831e..53517da0 100644 --- a/fastNLP/__init__.py +++ b/fastNLP/__init__.py @@ -52,6 +52,7 @@ __all__ = [ "SpanFPreRecMetric", "CMRC2018Metric", "ClassifyFPreRecMetric", + "ConfusionMatrixMetric", "Optimizer", "SGD", diff --git a/fastNLP/core/__init__.py b/fastNLP/core/__init__.py index acbbc535..89c6558d 100644 --- a/fastNLP/core/__init__.py +++ b/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