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

Correction for predict_no_weights

parent a03357d0
Branches
No related tags found
1 merge request!15Resolve "Adding new datasets"
...@@ -48,9 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest): ...@@ -48,9 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest):
result_omp = np.mean(omp_trees_predictions, axis=1) result_omp = np.mean(omp_trees_predictions, axis=1)
result_omp = (result_omp - 0.5) * 2 result_omp = np.array([-1 if pred < 0.5 else 1 for pred in result_omp])
print(result_omp)
return result_omp return result_omp
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment