Skip to content

Commit 5168d04

Browse files
committed
Add log_loss param sgd_clf in test_groupcrossvalidation
1 parent 50c25e2 commit 5168d04

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_estimator/test_estimator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def test_crossvalidation(self):
274274
y[:1000] = 1
275275

276276
# Try to fit a model
277-
cls = HyperoptEstimator(classifier=sgd_classifier("sgd", loss="log"), preprocessing=[])
277+
cls = HyperoptEstimator(classifier=sgd_classifier("sgd", loss="log_loss"), preprocessing=[])
278278
cls.fit(X, y, cv_shuffle=True, n_folds=5,
279279
kfolds_group=np.array([0]*500 + [1]*500 + [2]*500 + [3]*500)) # noqa: E226
280280

0 commit comments

Comments
 (0)