change test fliw

839506da · 张彦钊 · 2709e96b · 839506da
Commit 839506da authored 6 years ago by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

multi.py tensnsorflow/multi.py +2 -2

No files found.
--- a/tensnsorflow/multi.py
+++ b/tensnsorflow/multi.py
@@ -236,8 +236,8 @@ if __name__ == '__main__':

    # TODO 删除下面的测试写入
    df = spark.sql("select y,z from esmm_train_data limit 6000")
-    df.write.format("tfrecords").save(path="hdfs:///strategy/esmm/tr",mode="overwrite")
-    # .option("recordType", "Example")
+    df.write.format("tfrecords").option("recordType", "Example").save(path="hdfs:///strategy/esmm/tr/a.tfrecord",mode="overwrite")
+    #

    print("done")