Merge pull request #16543 from Genie-Liu/fix-metrics.Precision

fix fluid.metrics.Precision bug
6 years ago · 220190d5d2
parent b75a69bad6 0be1ce5850
commit 220190d5d2
1 changed files with 5 additions and 3 deletions
--- a/python/paddle/fluid/metrics.py
+++ b/python/paddle/fluid/metrics.py
@ -241,9 +241,11 @@ class Precision(MetricBase):
            raise ValueError("The 'preds' must be a numpy ndarray.")
        if not _is_numpy_(labels):
            raise ValueError("The 'labels' must be a numpy ndarray.")
-        sample_num = labels[0]
+        sample_num = labels.shape[0]
+        preds = np.rint(preds).astype("int32")
+
        for i in range(sample_num):
-            pred = preds[i].astype("int32")
+            pred = preds[i]
            label = labels[i]
            if label == 1:
                if pred == label: