https://discountedgerberprotectplus.blogspot.com/2024/04/pytorch-confusion-matrix-plot.html