Labels

#5
by kmailiaymen10 - opened

hi, can you put the meaning of labels (Label_0, Label_1, Label_2, ...)

kmailiaymen10 changed discussion title from Lables to Labels
Tunisia.AI org

Hi @kmailiaymen10
I haven't done that yet due to the model head having bias in the output head instead of just 2, this complicated things as between 4 classes now I couldn't figure out how to decode them as highlighted in https://huggingface.co/tunis-ai/TunBERT/discussions/3
image.png

I would appreciate it if you have any leads, or if you want to dig things up, you can find their repo at https://github.com/instadeepai/tunbert/tree/main
Else, I can give it an eye when I have some time.
If this is high priority for you please highlight it so i can move it up my agenda.
thanks πŸ€—

Tunisia.AI org

@kmailiaymen10
sorry for the much belated delai, i totally forgot that this existed.
it was not related to the bias parameter, you can find more details in the https://huggingface.co/tunis-ai/TunBERT/blob/main/modeling_tunbert.py#L31-L35
I will be closing this since this has been fixed πŸ€—

not-lain changed discussion status to closed

Sign up or log in to comment