add a more meaningful label names
#3
by not-lain - opened
change LABEL_1 and LABEL_0 to a more meaningful names
labels will be disabled as soon as we enable trust_remote_code
TODO: add custom config_class
not-lain changed discussion status to closed
not-lain changed discussion status to open
model.logits contains a list of 4 values, this is due to the original model implementation containing 2 neurons with bias = True
check pic for reference
not-lain changed discussion status to closed
@not-lain
So basically to use it effectivelly i need to finetune it on my specific dataet to get meaningful labels ?
It can not be used directly ?
@hamzabouajila i do advise you to finetune https://huggingface.co/google-bert/bert-base-uncased instead.
the implementation in this repo is somewhat broken , also the model head in this one has 2 layers which is a bit unconventional for transformer models.