修改测试文件

tensnsorflow/multi.py

@@ -36,7 +36,7 @@ def feature_engineer():
     validate_date = con_sql(db, sql)[0].values.tolist()[0]
     print("validate_date:" + validate_date)
     temp = datetime.datetime.strptime(validate_date, "%Y-%m-%d")
-    start = (temp - datetime.timedelta(days=3)).strftime("%Y-%m-%d")
+    start = (temp - datetime.timedelta(days=2)).strftime("%Y-%m-%d")
     print(start)
 
     sql = "select e.y,e.z,e.stat_date,e.ucity_id,feat.level2_ids,e.ccity_name,u.device_type,u.manufacturer," \
@@ -160,7 +160,7 @@ def get_predict(date,value_map,app_list_map,level2_map,level3_map):
 
     spark.createDataFrame(rdd.filter(lambda x: x[6] == 0)
                                    .map(lambda x: (x[0], x[1], x[2],x[9],x[10],x[11],x[12],x[13],x[14],x[15],
-                                                   x[16,x[17],x[18],x[19],x[20],x[21],x[22],x[23]]))) \
+                                                   x[16],x[17],x[18],x[19],x[20],x[21],x[22],x[23]))) \
         .toDF("app_list", "level2_ids", "level3_ids","ucity_id",
              "ccity_name", "device_type","manufacturer", "channel", "time", "hospital_id",
               "treatment_method", "price_min", "price_max", "treatment_time", "maintain_time",