Commit bf240b77 authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Add missing self in ensemble_selection_forest_regressor

parent 1194ee2f
......@@ -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)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment