diff --git a/code/bolsonaro/models/ensemble_selection_forest_regressor.py b/code/bolsonaro/models/ensemble_selection_forest_regressor.py index a20d8ec8bee5a124a10bf708ae176cf48796458e..50c1dc8570d04dcb4ee4825e67298f9ea79b861f 100644 --- a/code/bolsonaro/models/ensemble_selection_forest_regressor.py +++ b/code/bolsonaro/models/ensemble_selection_forest_regressor.py @@ -31,7 +31,7 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta): val_score = self._score_metric(estimator.predict(X_val), y_val) scores_list.append(val_score) - class_list = list(library) + class_list = list(self._library) m = np.argmax(np.asarray(scores_list)) self._ensemble_selected = [class_list[m]] temp_pred = class_list[m].predict(X_val)