Commit 2cd09bf7 authored by 张彦钊's avatar 张彦钊

ffm转化由四核改为八核

parent d5c92888
......@@ -69,7 +69,7 @@ def ffm_transform(data, test_number, validation_number):
print("Start ffm transform")
start = time.time()
ffm_train = multiFFMFormatPandas()
data = ffm_train.fit_transform(data, y='y',n=50000,processes=4)
data = ffm_train.fit_transform(data, y='y',n=50000,processes=8)
with open(DIRECTORY_PATH+"ffm.pkl", "wb") as f:
pickle.dump(ffm_train, f)
......
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