Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,16 @@ datasets:
|
|
7 |
language:
|
8 |
- ko
|
9 |
pipeline_tag: zero-shot-classification
|
10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
language:
|
8 |
- ko
|
9 |
pipeline_tag: zero-shot-classification
|
10 |
+
---
|
11 |
+
|
12 |
+
kor-nli 및 klue-nli로 학습하였습니다.
|
13 |
+
test accuracy(KLUE-NLI): 0.889667
|
14 |
+
|
15 |
+
```
|
16 |
+
model_name = 'deliciouscat/kf-deberta-base-cross-nli'
|
17 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
18 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3).to(device)
|
19 |
+
|
20 |
+
def encode(examples):
|
21 |
+
return tokenizer(examples['premise'], examples['hypothesis'], truncation=True, padding='max_length', max_length=128)
|
22 |
+
```
|