From 16139ce61292b9630f8b484e799d5a73158f03d1 Mon Sep 17 00:00:00 2001 From: leilei_snow Date: Tue, 28 Apr 2020 17:10:13 +0800 Subject: [PATCH] modify class EvaluationBase to private api. --- mindspore/nn/metrics/__init__.py | 2 -- mindspore/nn/metrics/{evaluation.py => _evaluation.py} | 0 mindspore/nn/metrics/accuracy.py | 2 +- mindspore/nn/metrics/precision.py | 2 +- mindspore/nn/metrics/recall.py | 2 +- 5 files changed, 3 insertions(+), 5 deletions(-) rename mindspore/nn/metrics/{evaluation.py => _evaluation.py} (100%) diff --git a/mindspore/nn/metrics/__init__.py b/mindspore/nn/metrics/__init__.py index 06429e200e..b82673b2ba 100755 --- a/mindspore/nn/metrics/__init__.py +++ b/mindspore/nn/metrics/__init__.py @@ -20,7 +20,6 @@ on the evaluation dataset. It's used to choose the best model. """ from .accuracy import Accuracy from .error import MAE, MSE -from .evaluation import EvaluationBase from .metric import Metric from .precision import Precision from .recall import Recall @@ -32,7 +31,6 @@ __all__ = [ "names", "get_metric_fn", "Accuracy", "MAE", "MSE", - "EvaluationBase", "Metric", "Precision", "Recall", diff --git a/mindspore/nn/metrics/evaluation.py b/mindspore/nn/metrics/_evaluation.py similarity index 100% rename from mindspore/nn/metrics/evaluation.py rename to mindspore/nn/metrics/_evaluation.py diff --git a/mindspore/nn/metrics/accuracy.py b/mindspore/nn/metrics/accuracy.py index f131432ddd..9537b4b95c 100644 --- a/mindspore/nn/metrics/accuracy.py +++ b/mindspore/nn/metrics/accuracy.py @@ -14,7 +14,7 @@ # ============================================================================ """Accuracy.""" import numpy as np -from .evaluation import EvaluationBase +from ._evaluation import EvaluationBase class Accuracy(EvaluationBase): diff --git a/mindspore/nn/metrics/precision.py b/mindspore/nn/metrics/precision.py index 633b9f8e2c..c0a3f9c12e 100644 --- a/mindspore/nn/metrics/precision.py +++ b/mindspore/nn/metrics/precision.py @@ -18,7 +18,7 @@ import sys import numpy as np from mindspore._checkparam import Validator as validator -from .evaluation import EvaluationBase +from ._evaluation import EvaluationBase class Precision(EvaluationBase): diff --git a/mindspore/nn/metrics/recall.py b/mindspore/nn/metrics/recall.py index da06321aa3..e0bd7538a3 100644 --- a/mindspore/nn/metrics/recall.py +++ b/mindspore/nn/metrics/recall.py @@ -18,7 +18,7 @@ import sys import numpy as np from mindspore._checkparam import Validator as validator -from .evaluation import EvaluationBase +from ._evaluation import EvaluationBase class Recall(EvaluationBase):