From 86c4cf105924e66286e0de936c8e7f27edc9c327 Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Fri, 6 Mar 2020 09:04:28 +0100
Subject: [PATCH] Fix wrong variable name in
ensemble_selection_forest_regressor
---
code/bolsonaro/models/ensemble_selection_forest_regressor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/code/bolsonaro/models/ensemble_selection_forest_regressor.py b/code/bolsonaro/models/ensemble_selection_forest_regressor.py
index 50c1dc8..1e63c8c 100644
--- a/code/bolsonaro/models/ensemble_selection_forest_regressor.py
+++ b/code/bolsonaro/models/ensemble_selection_forest_regressor.py
@@ -45,7 +45,7 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta):
temp_score = self._score_metric(temp_mean, y_val)
if (temp_score < best_score):
candidate_index = j
- best_score = tmp_score
+ best_score = temp_score
temp_pred = np.delete(temp_pred, -1, 0)
self._ensemble_selected.append(class_list[candidate_index])
temp_pred = np.vstack((temp_pred, class_list[candidate_index].predict(X_val)))
--
GitLab