From 31a58a03d3050120cbdd74aef72242b508014025 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:23:35 +0100
Subject: [PATCH] Added correction for edge case pred =
---
code/bolsonaro/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py
index cb1702d..fc289af 100644
--- a/code/bolsonaro/trainer.py
+++ b/code/bolsonaro/trainer.py
@@ -109,6 +109,7 @@ class Trainer(object):
y_pred = model.predict_no_weights(X)
if type(model) is OmpForestBinaryClassifier:
y_pred = np.sign(y_pred)
+ y_pred = np.where(y_pred==0, 1, y_pred)
result = self._classification_score_metric(y_true, y_pred)
return result
--
GitLab