add user

tensnsorflow/ffm.py

@@ -142,7 +142,8 @@ def get_data():
     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=6)).strftime("%Y-%m-%d")
+    start = (temp - datetime.timedelta(days=2)).strftime("%Y-%m-%d")
+    print(start)
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
     sql = "select e.device_id,e.y,e.z,e.stat_date,e.ucity_id,e.cid_id,e.clevel1_id,e.ccity_name," \
           "u.device_type,u.manufacturer,u.channel," \
@@ -199,15 +200,15 @@ def transform(a,validate_date):
 
 def get_predict_set(ucity_id, cid,model):
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
-    sql = "select e.device_id,e.y,e.z,e.stat_date,e.ucity_id,e.cid_id,e.clevel1_id,e.ccity_name,e.label," \
+    sql = "select e.device_id,e.y,e.z,e.stat_date,e.ucity_id,e.cid_id,e.clevel1_id,e.ccity_name," \
           "u.device_type,u.manufacturer,u.channel," \
           "home.jingxuan,home.zhibo,home.nose,home.eyes,home.weizheng,home.teeth,home.lunkuo," \
-          "home.meifu,home.xizhi,home.zhifang,home.longxiong,home.simi,home.maofa,home.gongli,home.korea " \
+          "home.meifu,home.xizhi,home.zhifang,home.longxiong,home.simi,home.maofa,home.gongli,home.korea,e.label " \
           "from esmm_pre_data e left join user_feature u on e.device_id = u.device_id " \
           "left join home_tab_click home on e.device_id = home.device_id"
     df = con_sql(db, sql)
     df = df.rename(columns={0: "device_id", 1: "y", 2: "z", 3: "stat_date", 4: "ucity_id", 5: "cid_id",
-                            6: "clevel1_id", 7: "ccity_name",8:"label"})
+                            6: "clevel1_id", 7: "ccity_name",26:"label"})
     df = df[df["cid_id"].isin(cid)]
     df = df[df["ucity_id"].isin(ucity_id)]
     print(df.shape)