修改测试文件

a50dda9a · 张彦钊 · 595b77d3 · a50dda9a
Commit a50dda9a authored Apr 25, 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
@@ -94,7 +94,8 @@ def feature_engineer():
    number = len(app_list_unique)
    app_list_map = dict(zip(app_list_unique, list(range(1, number + 1))))

-    print(df.select("app_list","stat_date").rdd.map(lambda x:(app_list_func(x[0],app_list_map),x[1])).first())
+    a = df.select("app_list","stat_date").rdd.map(lambda x:(app_list_func(x[0],app_list_map),x[1]))
+    spark.createDataFrame(a).show(6)