Multi-label Text Classification with BERT and PyTorch ...
curiousily.com › posts › multi-label-textTL;DR Learn how to prepare a dataset with toxic comments for multi-label text classification (tagging). We’ll fine-tune BERT using PyTorch Lightning and evaluate the model. Multi-label text classification (or tagging text) is one of the most common tasks you’ll encounter when doing NLP. Modern Transformer-based models (like BERT) make use of pre-training on vast amounts of text data that makes fine-tuning faster, use fewer resources and more accurate on small (er) datasets.