Skip to content

Commit 1d385ad

Browse files
authored
Merge pull request #438 from wangzhen38/wz38-stuff
xdeepfm_f
2 parents 3ef7a1c + b50e495 commit 1d385ad

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

models/rank/xdeepfm/static_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,9 @@ def net(self, input, is_infer=False):
8686

8787
predict_2d = paddle.concat(x=[1 - pred, pred], axis=1)
8888

89-
auc, batch_auc_var, _ = paddle.fluid.layers.auc(input=predict_2d,
90-
label=self.label_input,
91-
slide_steps=0)
89+
auc, batch_auc_var, _ = paddle.static.auc(input=predict_2d,
90+
label=self.label_input,
91+
slide_steps=0)
9292

9393
self.inference_target_var = auc
9494
if is_infer:

0 commit comments

Comments
 (0)