Add stacked ensemble to __init__ file for pipelines (#1326)

* test * switch order * test * add stacked ensembles to init * oops, revert * release notes
alteryx · Oct 21, 2020 · 55e1776 · 55e1776
1 parent d66d21a
commit 55e1776
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/docs/source/release_notes.rst b/docs/source/release_notes.rst
@@ -15,6 +15,7 @@ Release Notes
         * Fixed ML performance issue with ordered datasets: always shuffle data in automl's default CV splits :pr:`1265`
         * Fixed ``boosting type='rf'`` for LightGBM Classifier, as well as ``num_leaves`` error :pr:`1302`
         * Fixed bug in ``explain_predictions_best_worst`` where a custom index in the target variable would cause a ``ValueError`` :pr:`1318`
+        * Added stacked ensemble estimators to to ``evalml.pipelines.__init__`` file :pr:`1326`
     * Changes
         * Allow ``add_to_rankings`` to be called before AutoMLSearch is called :pr:`1250`
     * Documentation Changes

diff --git a/evalml/pipelines/__init__.py b/evalml/pipelines/__init__.py
@@ -22,7 +22,9 @@
     ExtraTreesClassifier,
     ExtraTreesRegressor,
     DecisionTreeClassifier,
-    DecisionTreeRegressor
+    DecisionTreeRegressor,
+    StackedEnsembleClassifier,
+    StackedEnsembleRegressor
 )
 
 from .pipeline_base import PipelineBase