https://wheyproteinonz.blogspot.com/2022/12/pytorch-confusion-matrix-plot.html
PyTorch Confusion Matrix Plot