Dataset for multilabel classification

[John6666]

I hope this is helpful…

Dataset label format for multi-label text classification

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.problem_type = "multi_label_classification", and define each label as a multi-hot vector (a list of 0/1 values, each corresponding to a different class). So for instance, given the task consists in assig…

Multi-Label Classification Model From Scratch: Step-by-Step Tutorial