diff --git a/fleetrec/models/ctr_dnn/model.py b/fleetrec/models/ctr_dnn/model.py index a025b8a15237e7f401a8a7293d03f55497d4ea06..8a1e2bbe59751f94f579a24abee91c65154f62af 100644 --- a/fleetrec/models/ctr_dnn/model.py +++ b/fleetrec/models/ctr_dnn/model.py @@ -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() diff --git a/setup.py b/setup.py index ca65bb3a64516368001cbaf7b9f30d074f7deeaf..37fa5355d111cf653dfbc5038881dde14618b59a 100644 --- a/setup.py +++ b/setup.py @@ -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"]