提交 7b3ab68c 编写于 作者: T tangwei

code clean

上级 2c77c937
......@@ -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
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册