change process argument

91605d0d · 张彦钊 · 89fdeab3 · 91605d0d
Commit 91605d0d authored Aug 17, 2018 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

processData.py processData.py +1 -1

No files found.
--- a/processData.py
+++ b/processData.py
@@ -75,7 +75,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=200000,processes=6)
+    data = ffm_train.fit_transform(data, y='y',n=100000,processes=6)
    with open(DIRECTORY_PATH+"ffm.pkl", "wb") as f:
        pickle.dump(ffm_train, f)