Skip to content
Snippets Groups Projects

WIP: Resolve "Adding new datasets"

Merged Charly Lamothe requested to merge 17-adding-new-datasets into master
2 files
+ 4
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -134,7 +134,7 @@ class SingleOmpForest(OmpForest):
Make all the base tree predictions
:param X: a Forest
:return: a np.array of the predictions of the entire forest
:return: a np.array of the predictions of the trees selected by OMP without applyong the weight
"""
forest_predictions = np.array([tree.predict(X) for tree in self._base_forest_estimator.estimators_])
Loading