fix anomaly detection docs

4 years ago · f1e1dcbb45
parent 2f2a10b8a3
commit f1e1dcbb45
1 changed files with 1 additions and 1 deletions
--- 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())