arne/EV5_Beeldherk_Bomen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Whoops all plataan gone

Browse Source
This commit is contained in:
Arne van Iterson 2023-10-22 20:24:55 +02:00
parent aed4d22199
commit 09f60efdb7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/suite.py
View File

@ -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])