From 09f60efdb75b887132964a2c51155636c13eb449 Mon Sep 17 00:00:00 2001 From: Arne van Iterson Date: Sun, 22 Oct 2023 20:24:55 +0200 Subject: [PATCH] Whoops all plataan gone --- src/suite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/suite.py b/src/suite.py index e458516..f892703 100644 --- a/src/suite.py +++ b/src/suite.py @@ -401,7 +401,7 @@ class CVSuite: tag_true = [guess[0] for guess in guesses ] tag_predict = [guess[1] for guess in guesses ] - labels = [Tree(tag).name for tag in range(0, 7)] + labels = [Tree(tag).name for tag in range(0, 8)] sns.heatmap(confusion_matrix(tag_true, tag_predict), xticklabels=labels, yticklabels=labels, ax=graph, annot=True, cbar=False, fmt='g') graph.set_title(self.models[modelnr][0])