From 7b3ab68ce60be07803cc4d12c7df1d54aa2c119a Mon Sep 17 00:00:00 2001 From: tangwei Date: Tue, 14 Apr 2020 13:20:08 +0800 Subject: [PATCH] code clean --- fleetrec/models/ctr_dnn/data_generator.py | 5 ----- fleetrec/trainer/transpiler_trainer.py | 2 +- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/fleetrec/models/ctr_dnn/data_generator.py b/fleetrec/models/ctr_dnn/data_generator.py index 3225cbd1..fd1ceff7 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 ebdfdefc..558c8a96 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) -- GitLab