update pyFM demo

csuldw · Feb 17, 2019 · 66bffed · 66bffed
1 parent 95802ac
commit 66bffed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Recommendation System/pyfm_demo.py b/Recommendation System/pyfm_demo.py
@@ -30,8 +30,8 @@ def load_data():
 X_train = v.fit_transform(X_train)
 X_test = v.transform(X_test)
 
-fm = pylibfm.FM(num_factors=1, 
-                num_iter=500, 
+fm = pylibfm.FM(num_factors=2, 
+                num_iter=200, 
                 verbose=True, 
                 task="classification", 
                 initial_learning_rate=0.001,