Skip to content
Snippets Groups Projects
Commit bf240b77 authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Add missing self in ensemble_selection_forest_regressor

parent 1194ee2f
Branches
No related tags found
2 merge requests!20Resolve "integration-sota",!19WIP: Resolve "Adding new datasets"
This commit is part of merge request !19. Comments created here will be created in the context of that merge request.
Show whitespace changes
Inline Side-by-side
code/bolsonaro/models/ensemble_selection_forest_regressor.py
+ 1
− 1
Edit View file @ bf240b77
...@@ -31,7 +31,7 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta): ...@@ -31,7 +31,7 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta):
val_score = self._score_metric(estimator.predict(X_val), y_val) val_score = self._score_metric(estimator.predict(X_val), y_val)
scores_list.append(val_score) scores_list.append(val_score)
class_list = list(library) class_list = list(self._library)
m = np.argmax(np.asarray(scores_list)) m = np.argmax(np.asarray(scores_list))
self._ensemble_selected = [class_list[m]] self._ensemble_selected = [class_list[m]]
temp_pred = class_list[m].predict(X_val) temp_pred = class_list[m].predict(X_val)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment