From 66bffed11fadef19b2d4996c5ef01b8de31cd51c Mon Sep 17 00:00:00 2001
From: csuldw <csu.ldw@csu.edu.cn>
Date: Sun, 17 Feb 2019 20:02:01 +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 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,