From ecb0d8180e971a6b14eb0932616ef4f62db53262 Mon Sep 17 00:00:00 2001
From: Alice BRENON <alice.brenon@ens-lyon.fr>
Date: Wed, 17 Jul 2024 12:47:26 +0200
Subject: [PATCH] Don't require the argument in getConfusionMatrix since it
doesn't make sense to pass one when loading a JSON
---
GEODE/Visualisation/ConfusionMatrix.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/GEODE/Visualisation/ConfusionMatrix.py b/GEODE/Visualisation/ConfusionMatrix.py
index b3c099c..47cec22 100644
--- a/GEODE/Visualisation/ConfusionMatrix.py
+++ b/GEODE/Visualisation/ConfusionMatrix.py
@@ -38,7 +38,7 @@ def confusionMatrix(data, labels):
matrix = confusion_matrix(truth, answers, labels=labels, normalize='true')
return {'matrix': matrix.tolist(), 'labels': labels}
-def getConfusionMatrix(inputFile, labels):
+def getConfusionMatrix(inputFile, labels=None):
if inputFile[-4:] == '.tsv':
return confusionMatrix(tabular(inputFile), labels)
elif inputFile[-5:] == '.json':
--
GitLab