File size: 568 Bytes
86967ee b12b4fe aa79f4e a00e093 aa79f4e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
---
library_name: transformers
license: mit
datasets:
- kor_nli
- klue
language:
- ko
pipeline_tag: zero-shot-classification
---
kor-nli 및 klue-nli로 학습하였습니다.
test accuracy(KLUE-NLI): 0.889667
```
model_name = 'deliciouscat/kf-deberta-base-cross-nli'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3).to(device)
def encode(examples):
return tokenizer(examples['premise'], examples['hypothesis'], truncation=True, padding='max_length', max_length=128)
``` |