Merge pull request PaddlePaddle#805 from wangzhen38/bug_fix1

update auc
muyunzhe · Jul 6, 2022 · 6b99585 · 6b99585
2 parents 0f1f4da + f5d2cc0
commit 6b99585
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/models/rank/deepfm/static_model.py b/models/rank/deepfm/static_model.py
@@ -81,8 +81,9 @@ def net(self, input, is_infer=False):
 
         predict_2d = paddle.concat(x=[1 - pred, pred], axis=1)
 
-        auc, batch_auc_var, _ = paddle.static.metric_op.auc(
-            input=predict_2d, label=self.label_input, slide_steps=0)
+        auc, batch_auc_var, _ = paddle.static.auc(input=predict_2d,
+                                                  label=self.label_input,
+                                                  slide_steps=0)
 
         self.inference_target_var = auc
         if is_infer: