Skip to content

F
ffm-baseline
Commit 3b34c96f authored by 张彦钊's avatar 张彦钊
Browse files
Options

修改测试文件

parent 12388704
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
tensnsorflow/multi.py
View file @ 3b34c96f
...@@ -105,8 +105,11 @@ def feature_engineer(): ...@@ -105,8 +105,11 @@ def feature_engineer():
value_map[x[13]], value_map[x[14]], value_map[x[15]], value_map[x[16]], value_map[x[13]], value_map[x[14]], value_map[x[15]], value_map[x[16]],
value_map[x[17]], x[18], x[19])) value_map[x[17]], x[18], x[19]))
spark.createDataFrame(test).write.csv('/recommend/va', mode='overwrite', header=True) # spark.createDataFrame(test).write.csv('/recommend/va', mode='overwrite', header=True)
spark.createDataFrame(train).write.csv('/recommend/tr', mode='overwrite', header=True) # spark.createDataFrame(train).write.csv('/recommend/tr', mode='overwrite', header=True)
a = spark.createDataFrame(train).toPandas
print(a.shape)
print("done") print("done")
rdd.unpersist() rdd.unpersist()
...@@ -239,6 +242,6 @@ if __name__ == '__main__': ...@@ -239,6 +242,6 @@ if __name__ == '__main__':
spark.sparkContext.setLogLevel("WARN") spark.sparkContext.setLogLevel("WARN")
validate_date, value_map, app_list_map, leve2_map, leve3_map = feature_engineer() 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) # get_predict(validate_date, value_map, app_list_map, leve2_map, leve3_map)
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