https://capressocoffemakerthecheapest.blogspot.com/2023/09/pytorch-confusion-matrix-plot.html