From 7423236627110c0c8cd42ccf70eea46b7965f22a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Lamercerie?=
<aurelien.lamercerie@tetras-libre.fr>
Date: Tue, 3 Oct 2023 17:34:02 +0200
Subject: [PATCH] update to handle zero division
---
ontoScorer/metric_score.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/ontoScorer/metric_score.py b/ontoScorer/metric_score.py
index 4a10bc2..646f90b 100644
--- a/ontoScorer/metric_score.py
+++ b/ontoScorer/metric_score.py
@@ -47,7 +47,7 @@ class Score:
self.y_true = y_true
self.y_pred = y_pred
if len(y_true) > 0:
- self.precision = precision_score(y_true, y_pred)
+ self.precision = precision_score(y_true, y_pred, zero_division=1)
self.recall = recall_score(y_true, y_pred)
self.f1 = f1_score(y_true, y_pred)
self.total_elements = len(y_true)
--
GitLab