Can I compute the class-wise metrics using the `compute()` function?

Hello everyone,

I am trying to figure out if it is possible to evaluate predictions of a custom BertForSequenceClassification model by class. Can the compute() function be used evaluate predictions for each class?

Thank you very much in advance!

Best regards,