diff --git a/code/bolsonaro/models/ensemble_selection_forest_regressor.py b/code/bolsonaro/models/ensemble_selection_forest_regressor.py index 50c1dc8570d04dcb4ee4825e67298f9ea79b861f..1e63c8cf3c34424e277d5a79d6f2349aee2aa5a5 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)))