Adapt sgd loss to newest scikit-learn version

automl · Jul 20, 2023 · 684afe3 · 684afe3
1 parent 79b5976
commit 684afe3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/2_multi_fidelity/2_sgd_datasets.py b/examples/2_multi_fidelity/2_sgd_datasets.py
@@ -89,7 +89,7 @@ def train(self, config: Configuration, instance: str, seed: int = 0) -> float:
 
             # SGD classifier using given configuration
             clf = SGDClassifier(
-                loss="log",
+                loss="log_loss",
                 penalty="elasticnet",
                 alpha=config["alpha"],
                 l1_ratio=config["l1_ratio"],

diff --git a/tests/fixtures/models.py b/tests/fixtures/models.py
@@ -71,7 +71,7 @@ def train(self, config: Configuration, instance: str = "0-1", budget: float = 1,
 
             # SGD classifier using given configuration
             clf = SGDClassifier(
-                loss="log",
+                loss="log_loss",
                 penalty="elasticnet",
                 alpha=config["alpha"],
                 l1_ratio=config["l1_ratio"],