Skip to content
Snippets Groups Projects

Resolve "Adding new datasets"

Merged Resolve "Adding new datasets"
17-adding-new-datasets into master
Merged Leo Bouscarrat requested to merge 17-adding-new-datasets into master
Compare
and
Show latest version
17 files
+ 29
− 8
Compare changes
  • Side-by-side
  • Inline
Files
17
code/bolsonaro/models/omp_forest.py
+ 2
− 2
@@ -33,8 +33,8 @@ class OmpForest(BaseEstimator, metaclass=ABCMeta):
# sklearn baseestimator api methods
def fit(self, X_forest, y_forest, X_omp, y_omp):
print(y_forest.shape)
print(set([type(y) for y in y_forest]))
# print(y_forest.shape)
# print(set([type(y) for y in y_forest]))
self._base_forest_estimator.fit(X_forest, y_forest)
self._extract_subforest(X_omp, y_omp) # type: OrthogonalMatchingPursuit
return self