Confusion matrix from 2 lists

Hey guys! I’m learning pytorch and I have to calculate the confusion matrix .The function for calculating confusion matrix accepts two lists as input parameters, predictions and truth. Can you suggest any ways for implementation?

Hi, I think this comment should be what you want.

1 Like