From f1e1dcbb45a344f7d0b0d53f8bf87678a38e4f3a Mon Sep 17 00:00:00 2001 From: zhangxinfeng3 Date: Thu, 24 Dec 2020 10:22:38 +0800 Subject: [PATCH] fix anomaly detection docs --- mindspore/nn/probability/toolbox/anomaly_detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/probability/toolbox/anomaly_detection.py b/mindspore/nn/probability/toolbox/anomaly_detection.py index ba92269b79..6f3aaca553 100644 --- a/mindspore/nn/probability/toolbox/anomaly_detection.py +++ b/mindspore/nn/probability/toolbox/anomaly_detection.py @@ -69,7 +69,7 @@ class VAEAnomalyDetection: sample_x (Tensor): The sample to be predicted, the shape is (N, C, H, W). Returns: - numpy.dtype, the predicted outlier score of the sample. + float, the predicted outlier score of the sample. """ reconstructed_sample = self.vae.reconstruct_sample(sample_x) return self._calculate_euclidean_distance(sample_x.asnumpy(), reconstructed_sample.asnumpy())