Skip to content
Snippets Groups Projects
Commit 0f1d25f7 authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Adding one [0] after nonzero

parent 24d53ff8
Branches master
No related tags found
1 merge request!15Resolve "Adding new datasets"
This commit is part of merge request !15. Comments created here will be created in the context of that merge request.
Show whitespace changes
Inline Side-by-side
code/bolsonaro/models/omp_forest.py
+ 2
− 2
Edit View file @ 0f1d25f7
...@@ -142,8 +142,8 @@ class SingleOmpForest(OmpForest): ...@@ -142,8 +142,8 @@ class SingleOmpForest(OmpForest):
forest_predictions /= self._forest_norms forest_predictions /= self._forest_norms
weights = self._omp.coef_ weights = self._omp.coef_
omp_trees_indices = np.nonzero(weights) omp_trees_indices = np.nonzero(weights)[0]
select_trees = np.mean(forest_predictions[omp_trees_indices], axis=0) select_trees = np.mean(forest_predictions[omp_trees_indices], axis=0)
print(len(omp_trees_indices))
return select_trees return select_trees
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment