search

abhishekkrthakur / bert-sentiment

MIT License
268 stars 103 forks source link

multi-label classification problem #7

Open sahelimukherjee92 opened 4 years ago

sahelimukherjee92 commented 4 years ago

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

mrvaruntandon commented 4 years ago

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 .

abhishekkrthakur commented 4 years ago

do you have a sample dataset?

sahelimukherjee92 commented 4 years ago

@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.