Whoops all plataan gone
This commit is contained in:
parent
aed4d22199
commit
09f60efdb7
@ -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])
|
||||||
|
Loading…
Reference in New Issue
Block a user