update label

peak_signal_to_noise_ratio.py

@@ -51,7 +51,7 @@ _CITATION = """
 
 
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
-class StructuralSimilarityIndexMeasure(evaluate.Metric):
+class PeakSignaltoNoiseRatio(evaluate.Metric):
     def _info(self):
         return evaluate.MetricInfo(
             description=_DESCRIPTION,
@@ -91,8 +91,8 @@ class StructuralSimilarityIndexMeasure(evaluate.Metric):
         sample_weight=None,
     ) -> Dict[str, float]:
         samples = zip(predictions, references)
-        return np.average(
-            list(map(lambda args: peak_signal_noise_ratio(*args, data_range), samples)), 
-            weights=sample_weight
-        )
+        psnrs = map(lambda args: peak_signal_noise_ratio(*args, data_range), samples)
+        return {
+            "Peak Signal-to-Noise Ratio": np.average(list(psnrs), weights=sample_weight)
+        }