提交 72c084b1 编写于 作者: F frankwhzhang

update model , test=develop

上级 33ea7c74
...@@ -69,6 +69,8 @@ def train_bpr_network(vocab_size, neg_size, hid_size, drop_out=0.2): ...@@ -69,6 +69,8 @@ def train_bpr_network(vocab_size, neg_size, hid_size, drop_out=0.2):
name="emb", name="emb",
initializer=fluid.initializer.XavierInitializer(), initializer=fluid.initializer.XavierInitializer(),
learning_rate=emb_lr_x)) learning_rate=emb_lr_x))
emb_src = fluid.layers.squeeze(input=emb_src, axes=[1])
emb_src_drop = fluid.layers.dropout(emb_src, dropout_prob=drop_out) emb_src_drop = fluid.layers.dropout(emb_src, dropout_prob=drop_out)
...@@ -134,6 +136,7 @@ def train_cross_entropy_network(vocab_size, neg_size, hid_size, drop_out=0.2): ...@@ -134,6 +136,7 @@ def train_cross_entropy_network(vocab_size, neg_size, hid_size, drop_out=0.2):
name="emb", name="emb",
initializer=fluid.initializer.XavierInitializer(), initializer=fluid.initializer.XavierInitializer(),
learning_rate=emb_lr_x)) learning_rate=emb_lr_x))
emb_src = fluid.layers.squeeze(input=emb_src, axes=[1])
emb_src_drop = fluid.layers.dropout(emb_src, dropout_prob=drop_out) emb_src_drop = fluid.layers.dropout(emb_src, dropout_prob=drop_out)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册