Whoops all plataan gone

This commit is contained in:
Arne van Iterson 2023-10-22 20:24:55 +02:00
parent aed4d22199
commit 09f60efdb7

View File

@ -401,7 +401,7 @@ class CVSuite:
tag_true = [guess[0] for guess in guesses ] tag_true = [guess[0] for guess in guesses ]
tag_predict = [guess[1] 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') 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]) graph.set_title(self.models[modelnr][0])