提交 759425ad 编写于 作者: Q Qiao Longfei

add is_sparse and is_distributed

上级 fb83c4b8
...@@ -15,8 +15,10 @@ def ctr_dnn_model(embedding_size, sparse_feature_dim): ...@@ -15,8 +15,10 @@ def ctr_dnn_model(embedding_size, sparse_feature_dim):
def embedding_layer(input): def embedding_layer(input):
return fluid.layers.embedding( return fluid.layers.embedding(
input=input, input=input,
is_sparse=True,
is_distributed=True,
size=[sparse_feature_dim, embedding_size], size=[sparse_feature_dim, embedding_size],
param_attr=fluid.ParamAttr(name="SparseFeatFactors", initializer=fluid.initializer.Normal(scale=1/math.sqrt(sparse_feature_dim)))) param_attr=fluid.ParamAttr(name="SparseFeatFactors", initializer=fluid.initializer.fluid.initializer.Uniform()))
sparse_embed_seq = map(embedding_layer, sparse_input_ids) sparse_embed_seq = map(embedding_layer, sparse_input_ids)
concated = fluid.layers.concat(sparse_embed_seq + [dense_input], axis=1) concated = fluid.layers.concat(sparse_embed_seq + [dense_input], axis=1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册