change test file

eda/esmm/Model_pipline/train.py

@@ -300,11 +300,7 @@ def main(_):
     FLAGS.model_dir = FLAGS.model_dir + FLAGS.dt_dir
     #FLAGS.data_dir  = FLAGS.data_dir + FLAGS.dt_dir
 
-    tr_files = glob.glob("/home/gmuser/test/part*")
-    random.shuffle(tr_files)
-    print("tr_files:", tr_files)
-
-    # tr_files = get_filename("tr")
+    tr_files = get_filename("tr")
     va_files = get_filename("va")
     te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]
 
@@ -374,8 +370,8 @@ if __name__ == "__main__":
     os.system(a)
     print("环境")
     print(os.system(a))
-    # path = "hdfs://172.16.32.4:8020/strategy/esmm/"
-    # tf.logging.set_verbosity(tf.logging.INFO)
-    # tf.app.run()
-    # print("耗时(分钟)：")
-    # print((time.time()-b)/60)
+    path = "hdfs://172.16.32.4:8020/strategy/esmm/"
+    tf.logging.set_verbosity(tf.logging.INFO)
+    tf.app.run()
+    print("耗时(分钟)：")
+    print((time.time()-b)/60)
\ No newline at end of file