diff --git a/fleetrec/models/ctr_dnn/data_generator.py b/fleetrec/models/ctr_dnn/data_generator.py index 3225cbd1b358b528b551ac21f62e0a56ae4bc3ab..fd1ceff711e38619ab42e51b0ba02ba4b4c80a6e 100644 --- a/fleetrec/models/ctr_dnn/data_generator.py +++ b/fleetrec/models/ctr_dnn/data_generator.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. from __future__ import print_function -from abc import ABC from fleetrec.reader.reader import Reader from fleetrec.utils import envs @@ -59,7 +58,3 @@ class TrainReader(Reader): yield zip(feature_name, [dense_feature] + sparse_feature + [label]) return reader - - -class EvaluateReader(Reader, ABC): - pass diff --git a/fleetrec/trainer/transpiler_trainer.py b/fleetrec/trainer/transpiler_trainer.py index ebdfdefc6b00b7f74d36fcbbd521f5cafd3a5312..558c8a96ccabc1fbe215dae05cb10af7b554d9fc 100644 --- a/fleetrec/trainer/transpiler_trainer.py +++ b/fleetrec/trainer/transpiler_trainer.py @@ -43,7 +43,7 @@ class TranspileTrainer(Trainer): batch_size = envs.get_global_env("batch_size", None, namespace) reader_class = envs.get_global_env("class", None, namespace) abs_dir = os.path.dirname(os.path.abspath(__file__)) - reader = os.path.join(abs_dir, '..', 'reader_implement.py') + reader = os.path.join(abs_dir, '../reader', 'reader_instance.py') pipe_cmd = "python {} {} {} {}".format(reader, reader_class, "TRAIN", self._config) train_data_path = envs.get_global_env("train_data_path", None, namespace)