Skip to content

F
ffm-baseline
Commit 7b4ecbe6 authored by 张彦钊's avatar 张彦钊
Browse files
Options

change transform

parent 9ba791f3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tensnsorflow/ffm.py
View file @ 7b4ecbe6
...@@ -68,9 +68,9 @@ def get_data(): ...@@ -68,9 +68,9 @@ def get_data():
def transform(df,validate_date): def transform(df,validate_date):
model = multiFFMFormatPandas() model = multiFFMFormatPandas()
for i in range(80000,200000,10000): for i in range(100000,200000,10000):
a = time.time() 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() b = time.time()
print("{}cost{}".format(i,b - a)) print("{}cost{}".format(i,b - a))
# df = pd.DataFrame(df) # df = pd.DataFrame(df)
...@@ -206,10 +206,10 @@ class multiFFMFormatPandas: ...@@ -206,10 +206,10 @@ class multiFFMFormatPandas:
x = 0 x = 0
while True: while True:
if x + step < data.__len__(): 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 x = x + step + 1
else: else:
data_list.append(data.iloc[x:data.__len__()]) data_list.append(data.loc[x:data.__len__()])
break break
return data_list return data_list
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment