https://discountsnowblowersthisinstant.blogspot.com/2023/11/pytorch-confusion-matrix-plot.html