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	208	January 20, 2025
Data format for BertForSequenceClassification with num_labels > 2 Beginners	5	4931	August 2, 2021
Multi-class Using Dataset 🤗Datasets	5	3716	February 10, 2023
Creating a Sequence of ClassLabel for multi-label and multi-class problems 🤗Datasets	5	758	March 26, 2024
Could someone please explain how to make a multi-label dataset from csv? Beginners	2	3591	May 31, 2022