test

eda/esmm/Model_pipline/train.py

@@ -382,11 +382,11 @@ def main(te_files):
 if __name__ == "__main__":
     b = time.time()
     path = "hdfs://172.16.32.4:8020/strategy/esmm/"
-    tf.logging.set_verbosity(tf.logging.INFO)
+    # tf.logging.set_verbosity(tf.logging.INFO)
     te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/test_nearby/part-r-00000"]
     print("hello up")
     result = main(te_files)
-    df = pd.DataFrame(result,columns=["sample_id","uid","city","cid_id","pctcvr"])
+    df = pd.DataFrame(result,columns=["uid","city","cid_id","pctcvr"])
     df.head(10)
     print("hello down")
     print("耗时(分钟)：")