deliciouscat commited on
Commit
aa79f4e
·
verified ·
1 Parent(s): b12b4fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -1
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
+ ```