Using distilbert-base-uncased-finetuned-sst-2-english for multiclass classification?
model = TFDistilBertForSequenceClassification.from_pretrained(‘ProsusAI/finbert’, num_labels=2)is working fine butmodel = TFDistilBertForSequenceClassification.from_pretrained(‘ProsusAI/finbert’, num_labels=5)is giving error.How can we finetune distilbert-base-uncased-finetuned-sst-2-english for multiclass(5 classes) classification?