https://discountedgerberprotectplus.blogspot.com/2024/04/pytorch-confusion-matrix-plot.html
Pytorch Confusion Matrix Plot