mathdatech/python-scripts
1
0
Fork 0
mirror of https://github.com/metafy-social/python-scripts.git synced 2024-11-23 20:11:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Confusion Matrix function

Browse Source
This commit is contained in:
lucas de oliveira carvalho 2022-10-11 09:23:53 -03:00
parent f9e93bf8ba
commit bd842f4a2b
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/Confusion_Matrix/make_confusion_matrix.py
View File

@ -51,9 +51,4 @@ def make_confusion_matrix(y_true, y_pred, classes=None, figsize=(10, 10), text_s
# Save the figure to the current working directory
if savefig:
fig.savefig("confusion_matrix.png")
y_true = [0, 0, 0, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1, 0]
y_pred = [0, 0, 1, 0, 1, 1, 0, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 1, 0]
make_confusion_matrix(y_true, y_pred)
fig.savefig("confusion_matrix.png")