Skip to content

F
ffm-baseline
Commit b5e8f403 authored by Your Name's avatar Your Name
Browse files
Options

change train.py

parent f845d959
gyz
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
eda/esmm/Model_pipline/train.py
View file @ b5e8f403
...@@ -421,8 +421,7 @@ if __name__ == "__main__": ...@@ -421,8 +421,7 @@ if __name__ == "__main__":
main(tr_files) main(tr_files)
elif FLAGS.task_type == 'infer': elif FLAGS.task_type == 'infer':
te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir] te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]
print(te_files[0]) queue_name = te_files[0].split('_')[-1].split('/')[0] + "_queue"
queue_name = te_files[0].split('_')[-1] + "_queue"
print(queue_name + " task") print(queue_name + " task")
result = main(te_files) result = main(te_files)
df = df_sort(result,queue_name) df = df_sort(result,queue_name)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment