delete time

tensnsorflow/ffm.py

@@ -200,8 +200,9 @@ def transform(a,validate_date):
     test = test.drop("stat_date",axis=1)
     print("train shape")
     print(train.shape)
-    train.to_csv(path + "tr.csv", sep="\t", index=False)
-    test.to_csv(path + "va.csv", sep="\t", index=False)
+
+    # train.to_csv(path + "tr.csv", sep="\t", index=False)
+    # test.to_csv(path + "va.csv", sep="\t", index=False)
 
     return model
 
@@ -291,7 +292,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)

tensnsorflow/test.py

@@ -65,10 +65,16 @@ def click():
         two_map[i] = df.loc[df[2] == i].shape[0] / n
     print(sorted(two_map.items(), key=lambda x: x[1],reverse=True))
 
+def get_cid():
+    db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
+    sql = "select distinct cid_id from esmm_train_data where device_id = '358035085192742' " \
+          "and stat_date >= '2018-12-03'"
+    df = con_sql(db, sql)[0].values.tolist()
+    print(",".join(df))
 
 
 
 if __name__ == "__main__":
-   click()
+   get_cid()