Skip to content
This repository was archived by the owner on Dec 6, 2023. It is now read-only.

Commit 191fb42

Browse files
committed
Make FM test_improve pass on win32/py2. A bit unstable.
1 parent a7adba2 commit 191fb42

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

polylearn/tests/test_factorization_machine.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,15 +142,13 @@ def check_improve(degree):
142142
est = FactorizationMachineRegressor(degree=degree, n_components=5,
143143
fit_lower=None, fit_linear=False,
144144
beta=0.0001, max_iter=5, tol=0,
145-
random_state=0,
146-
verbose=True, compute_loss=True)
145+
random_state=0)
147146
with warnings.catch_warnings():
148147
warnings.simplefilter("ignore")
149148
y_pred_5 = est.fit(X, y).predict(X)
150149
est.set_params(max_iter=10)
151150
y_pred_10 = est.fit(X, y).predict(X)
152151

153-
print(mean_squared_error(y, y_pred_10), mean_squared_error(y, y_pred_5))
154152
assert_less_equal(mean_squared_error(y, y_pred_10),
155153
mean_squared_error(y, y_pred_5),
156154
msg="More iterations do not improve fit.")

0 commit comments

Comments
 (0)