change transform

tensnsorflow/ffm.py

@@ -68,9 +68,9 @@ def get_data():
 
 def transform(df,validate_date):
     model = multiFFMFormatPandas()
-    for i in range(80000,200000,10000):
+    for i in range(100000,200000,10000):
         a = time.time()
-        df = model.fit_transform(df, y="y", n=i, processes=18)
+        temp = model.fit_transform(df, y="y", n=i, processes=18)
         b = time.time()
         print("{}cost{}".format(i,b - a))
     # df = pd.DataFrame(df)
@@ -206,10 +206,10 @@ class multiFFMFormatPandas:
         x = 0
         while True:
             if x + step < data.__len__():
-                data_list.append(data.iloc[x:x + step])
+                data_list.append(data.loc[x:x + step])
                 x = x + step + 1
             else:
-                data_list.append(data.iloc[x:data.__len__()])
+                data_list.append(data.loc[x:data.__len__()])
                 break
 
         return data_list