Skip to content
Snippets Groups Projects

WIP: Resolve "Adding new datasets"

Merged Charly Lamothe requested to merge 17-adding-new-datasets into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -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)))
Loading