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

Fix wrong variable name in ensemble_selection_forest_regressor

parent bf240b77
No related branches found
No related tags found
2 merge requests!20Resolve "integration-sota",!19WIP: Resolve "Adding new datasets"
......@@ -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)))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment