gg4ever commited on
Commit
ce58aca
1 Parent(s): 017ea89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
README.md CHANGED
@@ -1,3 +1,51 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ language:
4
+ - ko
5
+ metrics:
6
+ - sklearn-accuracy_score
7
+ datasets:
8
+ - kor_3i4k
9
+ pipeline_tag: text-classification
10
  ---
11
+
12
+ # intent-classification-korean
13
+
14
+ fine-tuned for 'klue/roberta-base'
15
+ used data : 'kor_3i4k'
16
+
17
+ ## How to Get Started with the Model
18
+
19
+ ```python
20
+ from transformers import TextClassificationPipeline
21
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
22
+
23
+ model_path = "gg4ever/intent-classifcation-korean"
24
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
25
+ model = AutoModelForSequenceClassification.from_pretrained(model_path)
26
+
27
+ text_classifier = TextClassificationPipeline(
28
+ tokenizer=tokenizer,
29
+ model=model.to('cpu'),
30
+ return_all_scores=True
31
+ )
32
+
33
+ # predict
34
+ text = "이름이 뭐에요?"
35
+
36
+ preds_list = text_classifier(text)
37
+ preds_list
38
+
39
+ ```
40
+
41
+ ### Training Hyperparameters
42
+
43
+ |hyperparameters|values|
44
+ |-----------------------------|-------|
45
+ |predict_with_generate|True|
46
+ |evaluation_strategy|"steps"|
47
+ |per_device_train_batch_size|32|
48
+ |per_device_eval_batch_size|32|
49
+ |num_train_epochs|3|
50
+ |learning_rate|4e-5|
51
+ |warmup_steps|1000|