fix: Fixing the random state of ElasticNetClassifier by default, to e…

…nsure reproduciblity. Also included elasticnet in reproducibility tests (#1374) Co-authored-by: Daniel Grindrod <[email protected]> Co-authored-by: Li Jiang <[email protected]>
microsoft · Oct 29, 2024 · 72881d3 · 72881d3
1 parent 69da685
commit 72881d3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/flaml/automl/model.py b/flaml/automl/model.py
@@ -2429,6 +2429,11 @@ def config2params(self, config: dict) -> dict:
 
     def __init__(self, task="regression", **config):
         super().__init__(task, **config)
+        self.params.update(
+            {
+                "random_state": config.get("random_seed", 10242048),
+            }
+        )
         assert self._task.is_regression(), "ElasticNet for regression task only"
         self.estimator_class = ElasticNet
 

diff --git a/test/automl/test_regression.py b/test/automl/test_regression.py
@@ -236,6 +236,7 @@ def test_multioutput():
     "estimator",
     [
         "catboost",
+        "enet",
         "extra_tree",
         "histgb",
         "kneighbor",
@@ -342,6 +343,7 @@ def test_reproducibility_of_catboost_regression_model():
     "estimator",
     [
         "catboost",
+        "enet",
         "extra_tree",
         "histgb",
         "kneighbor",
@@ -385,7 +387,6 @@ def test_reproducibility_of_underlying_regression_models(estimator: str):
             automl._state.X_train_all, automl._state.y_train_all, automl._state.kf, best_model.model, "regression"
         )
     )
-
     assert pytest.approx(val_loss_flaml) == reproduced_val_loss_underlying_model