Skip to content

Commit a420f95

Browse files
authored
Merge pull request #2135 from pavlin-policar/fix-sgd-defaults
SGD: Fix default loss parameter
2 parents db543d6 + e61230b commit a420f95

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

Orange/classification/sgd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ class SGDClassificationLearner(SklLearner):
1313
__returns__ = LinearModel
1414
preprocessors = SklLearner.preprocessors + [Normalize()]
1515

16-
def __init__(self, loss='squared_loss',penalty='l2', alpha=0.0001,
16+
def __init__(self, loss='hinge', penalty='l2', alpha=0.0001,
1717
l1_ratio=0.15,fit_intercept=True, n_iter=5, shuffle=True,
1818
epsilon=0.1, random_state=None, learning_rate='invscaling',
1919
eta0=0.01, power_t=0.25, warm_start=False, average=False,

Orange/regression/linear.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ class SGDRegressionLearner(LinearRegressionLearner):
8383
__wraps__ = skl_linear_model.SGDRegressor
8484
preprocessors = SklLearner.preprocessors + [Normalize()]
8585

86-
def __init__(self, loss='squared_loss',penalty='l2', alpha=0.0001,
86+
def __init__(self, loss='squared_loss', penalty='l2', alpha=0.0001,
8787
l1_ratio=0.15, fit_intercept=True, n_iter=5, shuffle=True,
8888
epsilon=0.1, n_jobs=1, random_state=None,
8989
learning_rate='invscaling', eta0=0.01, power_t=0.25,

Orange/tests/test_sgd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def setUpClass(cls):
3535
def test_SGDClassification(self):
3636
sgd = SGDClassificationLearner()
3737
res = CrossValidation(self.iris, [sgd], k=3)
38-
self.assertGreater(AUC(res)[0], 0.85)
38+
self.assertGreater(AUC(res)[0], 0.8)
3939

4040
def test_coefficients(self):
4141
lrn = SGDClassificationLearner()

0 commit comments

Comments
 (0)