Downloads · 30 days
10
2% of all-time downloads
brutusxu/roberta-large-finetuned-mnli
roberta-large-finetuned-mnli is a text classification model from brutusxu. Use it when you need a label for a piece of text. It is set up for transformers.
roberta-large finetuned on mnli dataset https://huggingface.co/datasets/multinli
Downloads · 30 days
10
2% of all-time downloads
All-time downloads
469
Public
Repo size
2.8 GB
Likes
0
Public
Click a slice to open those files.
.bin1.4 GB · 100%
From the Hugging Face model README
roberta-large finetuned on mnli dataset https://huggingface.co/datasets/multi_nli
accuracy on val_match: 0.9013 accuracy on val_mismatch: 0.8995
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained('roberta-large')
model = AutoModelForSequenceClassification
input_ids = tokenizer('I love English </s> I like English.',return_tensors='pt').input_ids
with torch.no_grad():
logits = model(input_ids).logits