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	243	January 20, 2025
Multi-class Using Dataset 🤗Datasets	5	3760	February 10, 2023
Creating a Sequence of ClassLabel for multi-label and multi-class problems 🤗Datasets	5	822	March 26, 2024
How to build a multi-label & multi-class dataset correctly? Beginners	4	910	April 18, 2025
Data format for BertForSequenceClassification with num_labels > 2 Beginners	5	4963	August 2, 2021