Update model.py

yapf checked

Update model.py
yapf checked
a3be0848 · Qdriving · GitHub · 5a3bd669 · a3be0848
隐藏空白更改
内联并排

Showing with 12 addition and 11 deletion

models/rank/dien/model.py models/rank/dien/model.py +12 -11

未找到文件。
--- a/models/rank/dien/model.py
+++ b/models/rank/dien/model.py
@@ -248,17 +248,18 @@ class Model(ModelBase):
                keep_dim=True)
            test_neg = fluid.layers.reduce_sum(
-                fluid.layers.log(
+                fluid.layers.reduce_sum(
-                    fluid.layers.sigmoid(
+                    fluid.layers.log(
-                        fluid.layers.reduce_sum(
+                        fluid.layers.sigmoid(
-                            gru_out_pad[:, start_value:seq_shape[1] - 1, :]
+                            fluid.layers.reduce_sum(
-                            * neg_seq_pad[:, start_value + 1:seq_shape[
+                                gru_out_pad[:, start_value:seq_shape[1] - 1, :]
-                                1], :],
+                                * neg_seq_pad[:, start_value + 1:seq_shape[
-                            dim=2,
+                                    1], :],
-                            keep_dim=True))),
+                                dim=2,
-                dim=2),
+                                keep_dim=True))),
-            dim=1,
+                    dim=2),
-            keep_dim=True)
+                dim=1,
+                keep_dim=True)
            aux_loss = fluid.layers.mean(test_neg + test_pos)