stanfordnlp/snli
Viewer • Updated • 570k • 31.6k • 92
How to use AntoineBlanot/roberta-nli with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("zero-shot-classification", model="AntoineBlanot/roberta-nli") # Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("AntoineBlanot/roberta-nli")
model = AutoModel.from_pretrained("AntoineBlanot/roberta-nli")This model is a fine-tuned model of roberta-large after being trained on a mixture of NLI datasets.
This model can classify a pair of sentence (a premise and a claim) into 3 classes:
This model can also be used for zero-shot classification tasks ! Please take a look at this repo for more information on zero-shot classification tasks.
This model has been trained in an efficient way and thus cannot be load directly from HuggingFace's hub. To use that model, please follow instructions on this repo.
For zero-shot classification tasks, please take a look at this repo.
| Data | Accuracy |
|---|---|
| MNLI (val. m) | 0.894 |
| MNLI (val. mm) | 0.895 |
| SNLI (val.) | 0.920 |
| SciTail (val.) | 0.934 |