bert-sentiment icon indicating copy to clipboard operation
bert-sentiment copied to clipboard

multi-label classification problem

Open sahelimukherjee92 opened this issue 4 years ago • 3 comments

how can this model be trained for a multi-label classification problem?

sahelimukherjee92 avatar May 01 '20 20:05 sahelimukherjee92

I am also interested in knowing how we can do multi-label classification through BERT. Does it depend on the 'out' attribute in the BERTBaseUncased Class? There is a hint in the video for this repo about changing loss functions for multi-label classification. Would like to know more about from @abhishekkrthakur .

mrvaruntandon avatar May 30 '20 14:05 mrvaruntandon

do you have a sample dataset?

abhishekkrthakur avatar Jun 15 '20 05:06 abhishekkrthakur

@abhishekkrthakur https://www.kaggle.com/c/jigsaw-toxic-comment-classification-challenge/data with different types of toxicity. How can we train a model for this type of problem? Thanks in advance.

sahelimukherjee92 avatar Jun 17 '20 14:06 sahelimukherjee92