add liuxao

tensnsorflow/ffm.py

@@ -156,7 +156,8 @@ def get_data():
     df = df.rename(columns={0: "y", 1: "z", 2: "stat_date", 3: "ucity_id",4: "clevel1_id", 5: "ccity_name",
                             6:"device_type",7:"manufacturer",8:"channel",9:"top",10:"time",11:"device_id"})
     print("esmm data ok")
-    print(df.head(2))
+    # print(df.head(2))
+
 
     df["clevel1_id"] = df["clevel1_id"].astype("str")
     df["y"] = df["y"].astype("str")
@@ -201,6 +202,8 @@ def transform(a,validate_date):
     print(train.shape)
     train.to_csv(path + "tr.csv", sep="\t", index=False)
     test.to_csv(path + "va.csv", sep="\t", index=False)
+    liuxiao = df[df["device_id"] == "358035085192742"]
+    liuxiao.to_csv(path + "liuxiao.csv", index=False)
 
     return model
 
@@ -210,7 +213,8 @@ def get_predict_set(ucity_id,model,ccity_name,manufacturer,channel):
     sql = "select e.y,e.z,e.label,e.ucity_id,e.clevel1_id,e.ccity_name," \
           "u.device_type,u.manufacturer,u.channel,c.top,cid_time.time,e.device_id,e.cid_id " \
           "from esmm_pre_data e left join user_feature u on e.device_id = u.device_id " \
-          "left join cid_type_top c on e.device_id = c.device_id left join cid_time on e.cid_id = cid_time.cid_id"
+          "left join cid_type_top c on e.device_id = c.device_id left join cid_time on e.cid_id = cid_time.cid_id " \
+          "where e.device_id = '358035085192742'"
     df = con_sql(db, sql)
     df = df.rename(columns={0: "y", 1: "z", 2: "label", 3: "ucity_id", 4: "clevel1_id", 5: "ccity_name",
                             6: "device_type", 7: "manufacturer", 8: "channel", 9: "top", 10: "time",
@@ -289,7 +293,7 @@ if __name__ == "__main__":
     a = time.time()
     temp, validate_date, ucity_id,ccity_name,manufacturer,channel = get_data()
     model = transform(temp, validate_date)
-    # get_predict_set(ucity_id,model,ccity_name,manufacturer,channel)
+    get_predict_set(ucity_id,model,ccity_name,manufacturer,channel)
     b = time.time()
     print("cost(分钟)")
     print((b-a)/60)