From 4451d8355ba34d842e114af4a90a8fbbd07776ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Thu, 5 Mar 2020 15:17:55 +0100
Subject: [PATCH] Correction for predict_no_weights
---
code/bolsonaro/models/omp_forest_classifier.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/code/bolsonaro/models/omp_forest_classifier.py b/code/bolsonaro/models/omp_forest_classifier.py
index a86e53b..31ca945 100644
--- a/code/bolsonaro/models/omp_forest_classifier.py
+++ b/code/bolsonaro/models/omp_forest_classifier.py
@@ -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
--
GitLab