From bf240b77f2f4698e77f958ba8f4ab2cc71164bd7 Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Fri, 6 Mar 2020 09:01:30 +0100
Subject: [PATCH] Add missing self 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 a20d8ec..50c1dc8 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)
--
GitLab