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
@@ -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)
Loading