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())