Skip to content

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

read tfrecords

parent a207254f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ ac40ecad
......@@ -180,9 +180,9 @@ if __name__ == "__main__":
spark.sparkContext.setLogLevel("WARN")
path = "hdfs://172.16.32.4:8020/strategy/esmm/"
df = spark.read.format("tfrecords").load(path+"nearby/part-r-00000")
df.show()
b = time.time()
tf.logging.set_verbosity(tf.logging.INFO)
tf.app.run()
print("耗时(分钟)：")
print((time.time()-b)/60)
\ No newline at end of file
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