klue-roberta-base-kornli
This model trained with Korean dataset. Input premise sentence and hypothesis sentence. You can use English, but don't expect accuracy. If the context is longer than 1200 characters, the context may be cut in the middle and the result may not come out well.
KLUE-RoBERTa-base-KorNLI DEMO: Ainize DEMO
KLUE-RoBERTa-base-KorNLI API: Ainize API
Overview
Language model: klue/roberta-base
Language: Korean
Training data: kakaobrain KorNLI
Eval data: kakaobrain KorNLI
Code: See Ainize Workspace
Usage
In Transformers
from transformers import AutoTokenizer, pipeline
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/klue-roberta-base-kornli")
classifier = pipeline(
"text-classification",
model="ehdwns1516/klue-roberta-base-kornli",
return_all_scores=True,
)
premise = "your premise"
hypothesis = "your hypothesis"
result = dict()
result[0] = classifier(premise + tokenizer.sep_token + hypothesis)[0]