WIP: Resolve "Adding new datasets"

code/bolsonaro/models/ensemble_selection_forest_regressor.py

@@ -53,14 +53,14 @@ class EnsembleSelectionForestRegressor(BaseEstimator, metaclass=ABCMeta):
 
     def score(self, X, y):
         predictions = self._predict_base_estimator(X)
-        mean_predictions = np.mean(predictions, axis=0)
-        return self._score_metric(mean_predictions, y)
+        return self._score_metric(predictions, y)
 
     def predict_base_estimator(self, X):
         predictions = list()
         for tree in self._ensemble_selected:
             predictions.append(tree.predict(X))
-        return np.array(predictions)
+        mean_predictions = np.mean(np.array(predictions), axis=0)
+        return mean_predictions
 
     @staticmethod
     def generate_library(X_train, y_train, random_state=None):