Hi, I’ve been able to train a multi-label Bert classifier using a custom Dataset object and the Trainer API from Transformers. The Dataset contains two columns: text and label. After tokenizing, I have all the needed columns for training. For multi-label classification I also set model.config.prob…

Dataset label format for multi-label text classification

trasformernewbe February 9, 2023, 6:35pm 10

Topic		Replies	Views
Dataset for multilabel classification 🤗Transformers	1	107	January 20, 2025
Could someone please explain how to make a multi-label dataset from csv? Beginners	2	3520	May 31, 2022
Data format for BertForSequenceClassification with num_labels > 2 Beginners	5	4895	August 2, 2021
BERT for Dataset with two label columns Beginners	1	435	January 22, 2024
BERT Multilabel - Different Training Dataset For Each Label? Intermediate	3	1289	December 27, 2021