change test file

c3ad5e63 · 张彦钊 · 1097e6d8 · c3ad5e63
Commit c3ad5e63 authored Apr 30, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/tensnsorflow/multi.py
+++ b/tensnsorflow/multi.py
@@ -201,9 +201,10 @@ def test():
    sql = "select stat_date,cid_id from esmm_train_data e where stat_date = '{}' limit 60".format("2019-04-25")

    df = spark.sql(sql)
+    df.show(6)

    # df.write.csv('/recommend/tr', mode='overwrite', header=True)
-    df.select("stat_date", "cid_id").write.format("avro").save("/recommend/tr/avro.csv")
+    df.write.format("avro").save("/recommend/tr")
    # df.write.format("avro").save("/recommend/tr/avro.csv")