change test file

92807d12 · 张彦钊 · 9d8eeae0 · 92807d12
Commit 92807d12 authored May 24, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

feature_engineering.py tensnsorflow/feature_engineering.py +3 -1

No files found.
--- a/tensnsorflow/feature_engineering.py
+++ b/tensnsorflow/feature_engineering.py
@@ -323,7 +323,7 @@ def get_predict(date,value_map,app_list_map,leve2_map,leve3_map):
        .save(path=path + "nearby/", mode="overwrite")
    print("nearby tfrecord done")

-    rdd.unpersist()
+



@@ -347,6 +347,8 @@ if __name__ == '__main__':
    validate_date, value_map, app_list_map, leve2_map, leve3_map = feature_engineer()
    get_predict(validate_date, value_map, app_list_map, leve2_map, leve3_map)

+    spark.stop()
+