From 95802acb59603ac1bc3885e3db7d50715e58bca3 Mon Sep 17 00:00:00 2001 From: csuldw Date: Sun, 17 Feb 2019 19:39:07 +0800 Subject: [PATCH] update pyFM demo --- Recommendation System/pyfm_demo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Recommendation System/pyfm_demo.py b/Recommendation System/pyfm_demo.py index 3787809..1ea5d9e 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=100, - num_iter=200, +fm = pylibfm.FM(num_factors=1, + num_iter=500, verbose=True, task="classification", initial_learning_rate=0.001,