修改测试文件

tensnsorflow/multi.py

@@ -100,7 +100,7 @@ def feature_engineer():
     spark.createDataFrame(test).toDF("app_list","level2_ids","level3_ids","stat_date","ucity_id", "ccity_name", "device_type", "manufacturer",
                   "channel", "top", "time", "hospital_id","treatment_method", "price_min",
                   "price_max", "treatment_time","maintain_time", "recover_time","y","z")\
-        .write.format("tfrecords").option("recordType", "Example").save(path=path+"va/", mode="overwrite")
+        .repartition(1).write.format("tfrecords").option("recordType", "Example").save(path=path+"va/", mode="overwrite")
 
     print("va write done")
     spark.createDataFrame(train).toDF("app_list","level2_ids","level3_ids","stat_date","ucity_id", "ccity_name", "device_type", "manufacturer",