Deep Learning, Machine Learning, Pytorch Lightning, Metrics, AI.
confusion matrix was computed by summing across different labels, which is unwanted. The metric will now match sklearns implementation if the new argument multilabel is set to True, meaning that it will output a binary confusion matrix per label. Confusion matrix now better support multi label