diff --git a/Recommendation System/pyfm_demo.py b/Recommendation System/pyfm_demo.py index 1ea5d9e..7c634f5 100644 --- 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,