change transform

tensnsorflow/ffm.py

@@ -68,7 +68,7 @@ def get_data():
 
 def transform(df,validate_date):
     model = multiFFMFormatPandas()
-    for i in range(100000,200000,10000):
+    for i in [200000,160000,130000]:
         a = time.time()
         temp = model.fit_transform(df, y="y", n=i, processes=18)
         b = time.time()
@@ -122,6 +122,15 @@ def get_statistics():
     df = df.drop("total", axis=1)
     return df
 
+def get_predict_set():
+    db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
+    sql = "select device_id,y,z,stat_date,ucity_id,cid_id,clevel1_id,ccity_name from esmm_pre_data where label = 0"
+    native_pre = con_sql(db, sql)
+    native_pre = native_pre.rename(columns={0: "device_id", 1: "y", 2: "z", 3: "stat_date", 4: "ucity_id", 5: "cid_id",
+                            6: "clevel1_id", 7: "ccity_name"})
+    print("native_pre ok")
+
+
 
 class multiFFMFormatPandas:
     def __init__(self):