Commit 0f1d25f7 authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Adding one [0] after nonzero

parent 24d53ff8
......@@ -142,8 +142,8 @@ class SingleOmpForest(OmpForest):
forest_predictions /= self._forest_norms
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)
return select_trees
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment