Hi I have a dataset for multi-label classification, I have 14 categories and a decent amount of data. However I’m having issues trying to get the training going. I have tried one hot encoding or passing the list of classes but keep getting errors. Basically I have a dataset that looks like this: {…

I hope this is helpful… [image] Dataset label format for multi-label text classification 🤗Datasets 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 lab…

Dataset for multilabel classification

🤗Transformers

John6666 January 20, 2025, 10:42pm 2

I hope this is helpful…

Topic		Replies	Views
Dataset label format for multi-label text classification 🤗Datasets	9	13646	February 9, 2023
Multi-class Using Dataset 🤗Datasets	5	3806	February 10, 2023
Data format for BertForSequenceClassification with num_labels > 2 Beginners	5	4999	August 2, 2021
Could someone please explain how to make a multi-label dataset from csv? Beginners	2	3660	May 31, 2022
Multilabel text classification Trainer API Beginners	8	22953	August 2, 2023

Dataset for multilabel classification

Related topics