Change the default value to LinearRegression(positive=True) in the …

…constructor of `StackingEnsemble` (#1238)
tinkoff-ai · Apr 21, 2023 · f06cd82 · f06cd82 · github-actions · Apr 21, 2023
1 parent 3cd565b
commit f06cd82
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,8 +12,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 -
 - 
 ### Changed
--
--
+- Set the default value of `final_model` to `LinearRegression(positive=True)` in the constructor of `StackingEnsemble` ([#1238](https://github.com/tinkoff-ai/etna/pull/1238))
+- 
 -
 -
 ### Fixed

diff --git a/etna/ensembles/stacking_ensemble.py b/etna/ensembles/stacking_ensemble.py
@@ -92,7 +92,7 @@ def __init__(
         """
         self._validate_pipeline_number(pipelines=pipelines)
         self.pipelines = pipelines
-        self.final_model = LinearRegression() if final_model is None else final_model
+        self.final_model = LinearRegression(positive=True) if final_model is None else final_model
         self._validate_backtest_n_folds(n_folds)
         self.n_folds = n_folds
         self.features_to_use = features_to_use