Commit 4451d835 authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Correction for predict_no_weights

parent a03357d0
......@@ -48,9 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest):
result_omp = np.mean(omp_trees_predictions, axis=1)
result_omp = (result_omp - 0.5) * 2
print(result_omp)
result_omp = np.array([-1 if pred < 0.5 else 1 for pred in result_omp])
return result_omp
......
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