提交 3045ed26 编写于 作者: T tangwei

fix import

上级 2cba27a6
......@@ -45,7 +45,7 @@ class Model(ModelBase):
return dense_input_var
def label_input():
label = fluid.layers.data(name="label", shape=[1], dtype="int64")
label = fluid.layers.data(name="click", shape=[1], dtype="int64")
return label
self.sparse_inputs = sparse_inputs()
......
......@@ -5,7 +5,7 @@ setup for fleet-rec.
from setuptools import setup
packages = ["fleetrec", "fleetrec.models", "fleetrec.models.ctr_dnn",
"fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in",
"fleetrec.examples", "fleetrec.examples.user_define",
"fleetrec.core", "fleetrec.core.engine",
"fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册