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

Commit 292512b

Browse files
committed
Make P-like arrays F-order in adagrad
1 parent c97b225 commit 292512b

13 files changed

+4809
-1886
lines changed

benchmarks/bench_20newsgroups.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,25 +16,29 @@
1616

1717

1818
estimators = {
19-
'fm-2': FactorizationMachineClassifier(n_components=30,
19+
'fm-2': FactorizationMachineClassifier(beta=0.0001,
20+
n_components=30,
2021
fit_linear=False,
2122
fit_lower=None,
2223
degree=2,
2324
random_state=0,
24-
max_iter=10),
25+
max_iter=20,
26+
tol=1e-10),
2527

26-
'polynet-2': PolynomialNetworkClassifier(n_components=15, degree=2,
28+
'polynet-2': PolynomialNetworkClassifier(beta=0.0001,
29+
n_components=15,
30+
degree=2,
2731
fit_lower=None,
28-
max_iter=10,
29-
random_state=0)
32+
max_iter=20,
33+
random_state=0,
34+
tol=1e-10)
3035
}
3136

3237
estimators['fm-3'] = clone(estimators['fm-2']).set_params(degree=3)
3338
estimators['fm-2-ada'] = clone(estimators['fm-2']).set_params(
34-
solver='adagrad', learning_rate=0.01, max_iter=20)
39+
solver='adagrad')
3540
estimators['fm-3-ada'] = clone(estimators['fm-3']).set_params(
36-
solver='adagrad', learning_rate=0.01, max_iter=20
37-
)
41+
solver='adagrad')
3842
estimators['polynet-3'] = (clone(estimators['polynet-2'])
3943
.set_params(degree=3, n_components=10))
4044

@@ -68,7 +72,7 @@
6872
for name, clf in sorted(estimators.items()):
6973
print("Training %s ... " % name, end="")
7074
if 'ada' in name:
71-
X_train, X_teest = X_train_csr, X_test_csr
75+
X_train, X_test = X_train_csr, X_test_csr
7276
else:
7377
X_train, X_test = X_train_csc, X_test_csc
7478
t0 = time()

0 commit comments

Comments
 (0)