修改测试文件

tensnsorflow/multi.py

@@ -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[17]], x[18], x[19]))
 
-    spark.createDataFrame(test).write.csv('/recommend/va', mode='overwrite', header=True)
-    spark.createDataFrame(train).write.csv('/recommend/tr', 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)
+
+    a = spark.createDataFrame(train).toPandas
+    print(a.shape)
     print("done")
     rdd.unpersist()
 
@@ -239,6 +242,6 @@ if __name__ == '__main__':
     spark.sparkContext.setLogLevel("WARN")
 
     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)