From 7be0625a85375270e8ee1ffe449f713136c3c2da Mon Sep 17 00:00:00 2001 From: Jiaqi Date: Tue, 30 Mar 2021 17:09:33 +0800 Subject: [PATCH] fix api bugs --- mindspore/nn/loss/loss.py | 4 ++-- mindspore/nn/metrics/bleu_score.py | 2 +- mindspore/nn/metrics/confusion_matrix.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/nn/loss/loss.py b/mindspore/nn/loss/loss.py index 93ab968df1..5be62467e3 100644 --- a/mindspore/nn/loss/loss.py +++ b/mindspore/nn/loss/loss.py @@ -452,7 +452,7 @@ class DiceLoss(_Loss): single_dice_coeff = (2 * intersection) / (unionset + self.smooth) dice_loss = 1 - single_dice_coeff - return dice_loss.mean() + return dice_loss @constexpr @@ -1078,7 +1078,7 @@ class FocalLoss(_Loss): >>> focalloss = nn.FocalLoss(weight=Tensor([1, 2]), gamma=2.0, reduction='mean') >>> output = focalloss(predict, target) >>> print(output) - 1.6610543 + 0.12516622 """ def __init__(self, weight=None, gamma=2.0, reduction='mean'): diff --git a/mindspore/nn/metrics/bleu_score.py b/mindspore/nn/metrics/bleu_score.py index 001401ddba..b5feb74c78 100644 --- a/mindspore/nn/metrics/bleu_score.py +++ b/mindspore/nn/metrics/bleu_score.py @@ -38,7 +38,7 @@ class BleuScore(Metric): >>> metric.clear() >>> metric.update(candidate_corpus, reference_corpus) >>> bleu_score = metric.eval() - >>> print(output) + >>> print(bleu_score) 0.5946035575013605 """ def __init__(self, n_gram=4, smooth=False): diff --git a/mindspore/nn/metrics/confusion_matrix.py b/mindspore/nn/metrics/confusion_matrix.py index c5fdabc18f..1d44d0c3ef 100644 --- a/mindspore/nn/metrics/confusion_matrix.py +++ b/mindspore/nn/metrics/confusion_matrix.py @@ -182,7 +182,7 @@ class ConfusionMatrixMetric(Metric): >>> y = Tensor(np.array([[[0], [1]], [[1], [0]]])) >>> avg_output = metric.eval() >>> print(avg_output) - [0.75] + [0.5] """ def __init__(self, skip_channel=True,