mathdatech/python-scripts
1
0
Fork 0
mirror of https://github.com/metafy-social/python-scripts.git synced 2025-02-17 13:58: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

5
scripts/Confusion_Matrix/make_confusion_matrix.py
View File

@ -52,8 +52,3 @@ 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)