textattack

/

bert-base-uncased-snli

like 0

Follow

TextAttack 35

I'm new to huggingface - unsure this is posted somewhere else, but here's the label mapping for this model:

TEXT_ATTACK_NUM_TO_LABEL_MAP = {
0: "contradiction",
1: "entailment",
2: "neutral"
}

Model runs great - but the mapping is off from the standard SNLI dataset labels of:

LABEL_TO_NUM_MAP = {
"entailment": 0,
"neutral": 1,
"contradiction": 2
}