Skip to content

P
PaddleRec

PaddlePaddle / PaddleRec

通知 68
Star 12
Fork 5
P
PaddleRec
未验证 提交 b8589509 编写于 作者: Q Qdriving 提交者： GitHub
浏览文件
操作

Update model.py 

yapf check
上级 d8282814
显示空白变更内容
内联 并排
Showing with 5 addition and 7 deletion
models/rank/dien/model.py
浏览文件 @ b8589509
...@@ -230,7 +230,7 @@ class Model(ModelBase): ...@@ -230,7 +230,7 @@ class Model(ModelBase):
pad_value) pad_value)
seq_shape = fluid.layers.shape(pos_seq_pad) seq_shape = fluid.layers.shape(pos_seq_pad)
if seq_shape[1] < 2: if (seq_shape[1] == 1):
aux_loss = 0 aux_loss = 0
else: else:
test_pos = fluid.layers.reduce_sum( test_pos = fluid.layers.reduce_sum(
...@@ -239,28 +239,26 @@ class Model(ModelBase): ...@@ -239,28 +239,26 @@ class Model(ModelBase):
fluid.layers.sigmoid( fluid.layers.sigmoid(
fluid.layers.reduce_sum( fluid.layers.reduce_sum(
gru_out_pad[:, start_value:seq_shape[1] - 1, :] gru_out_pad[:, start_value:seq_shape[1] - 1, :]
* pos_seq_pad[:, start_value + 1:seq_shape[ * pos_seq_pad[:, start_value +
1], :], 1:seq_shape[1], :],
dim=2, dim=2,
keep_dim=True))), keep_dim=True))),
dim=2), dim=2),
dim=1, dim=1,
keep_dim=True) keep_dim=True)
test_neg = fluid.layers.reduce_sum( test_neg = fluid.layers.reduce_sum(
fluid.layers.reduce_sum( fluid.layers.reduce_sum(
fluid.layers.log( fluid.layers.log(
fluid.layers.sigmoid( fluid.layers.sigmoid(
fluid.layers.reduce_sum( fluid.layers.reduce_sum(
gru_out_pad[:, start_value:seq_shape[1] - 1, :] gru_out_pad[:, start_value:seq_shape[1] - 1, :]
* neg_seq_pad[:, start_value + 1:seq_shape[ * neg_seq_pad[:, start_value +
1], :], 1:seq_shape[1], :],
dim=2, dim=2,
keep_dim=True))), keep_dim=True))),
dim=2), dim=2),
dim=1, dim=1,
keep_dim=True) keep_dim=True)
aux_loss = fluid.layers.mean(test_neg + test_pos) aux_loss = fluid.layers.mean(test_neg + test_pos)
# ------------------------- Interest Evolving Layer (GRU with attentional input (AIGRU)) -------------------------- # ------------------------- Interest Evolving Layer (GRU with attentional input (AIGRU)) --------------------------
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或