So data.c automatically takes the correct value (if there are 3 label columns then it’s 3), so there is no need to modify text_classifier_learner which is good! (I’ll edit the post above.)
I am not sure about the loss function, though, because printing its type only returns the super class FlattenedLoss, which could be either. Is there a more proper way to check its type?