提交 cd2b6919 编写于 作者: Q Qiao Longfei

fix sparse_feature_num

上级 48a76c01
......@@ -9,7 +9,7 @@ def ctr_dnn_model(embedding_size, sparse_feature_dim, extend_id_range=False):
name="dense_input", shape=[dense_feature_dim], dtype='float32')
sparse_feature_num = 26
if extend_id_range:
sparse_feature_num = 26 + 26 * 25
sparse_feature_num = 26 + 26 * 25 / 2
sparse_input_ids = [
fluid.layers.data(
name="C" + str(i), shape=[1], lod_level=1, dtype='int64')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册