search

UBC-NLP / marbert

UBC ARBERT and MARBERT Deep Bidirectional Transformers for Arabic
100 stars 15 forks source link

Multi label (multi-hot-encoded) in Fine-Tuning #3

Closed NoraAlt closed 2 years ago

NoraAlt commented 3 years ago

Hello,

I am following the example here, to fine-tune the MARBERT for Multi-label Classification.

My dataset looks like:

screenshot

Label_list=['Optimistic','Thankful','Empathetic','Pessimistic','Anxious','Sad','Annoyed','Denial','Official','Surprise','Joking']

I could not get how to pass the labels.. What I should pass for the "label_col" variable?

Any ideas?

Would greatly appreciate your help!