Skip to content

Commit 66bffed

Browse files
committed
update pyFM demo
1 parent 95802ac commit 66bffed

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Recommendation System/pyfm_demo.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ def load_data():
3030
X_train = v.fit_transform(X_train)
3131
X_test = v.transform(X_test)
3232

33-
fm = pylibfm.FM(num_factors=1,
34-
num_iter=500,
33+
fm = pylibfm.FM(num_factors=2,
34+
num_iter=200,
3535
verbose=True,
3636
task="classification",
3737
initial_learning_rate=0.001,

0 commit comments

Comments
 (0)