add data

tensnsorflow/ffm.py

@@ -47,8 +47,10 @@ def get_data():
     print(df.head())
     print("shape")
     print(df.shape)
-    device_tuple = tuple(df["device_id"].values.tolist())
-    df = pd.merge(df,get_statistics(device_tuple),on="device_id",how='left').fillna(0)
+    device_tuple = tuple(set(df["device_id"].values.tolist()))
+    a = device_tuple[0:2]
+    print(a)
+    df = pd.merge(df,get_statistics(a),on="device_id",how='left').fillna(0)
     print("merge")
     print(df.head())
     print("shape")
@@ -99,6 +101,7 @@ def get_statistics(device_tuple):
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='eagle')
     sql = "select device_id,device_type,channel,total,精选,直播,鼻部,眼部,微整,牙齿,轮廓,美肤抗衰," \
           "吸脂,脂肪填充,隆胸,私密,毛发管理,公立,韩国 from home_tab_click where device_id in {}".format(device_tuple)
+    print(sql)
     df = con_sql(db, sql)
     df = df.rename(columns={0:"device_id",1:"device_type",2:"channel",3:"total"})
     for i in df.columns.difference(["device_id", "device_type","channel","total"]):