0.6.11 a26

winedarksea · Apr 4, 2024 · 71bbe30 · 71bbe30
1 parent baf7d7c
commit 71bbe30
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/autots/evaluator/auto_ts.py b/autots/evaluator/auto_ts.py
@@ -1593,9 +1593,14 @@ def _return_best_model(
             metric_weighting = self.metric_weighting
         if template_cols is None:
             template_cols = self.template_cols_id
-        hens_model_results = self.initial_results.model_results[
-            self.initial_results.model_results['Ensemble'] == 2
-        ].copy()
+        if self.horizontal_ensemble_validation:
+            hens_model_results = self.validation_results.model_results[
+                self.validation_results.model_results['Ensemble'] == 2
+            ].copy()
+        else:
+            hens_model_results = self.initial_results.model_results[
+                self.initial_results.model_results['Ensemble'] == 2
+            ].copy()
         # remove failures
         hens_model_results = hens_model_results[
             hens_model_results['Exceptions'].isnull()