jeonseongju commited on
Commit
dfa96d1
โ€ข
1 Parent(s): aa07fee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -9,4 +9,19 @@
9
  * train_runtime: 1477.3876
10
  * train_steps_per_second: 2.416
11
  * train_loss: 0.3722160959110207
12
- * epoch: 5.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  * train_runtime: 1477.3876
10
  * train_steps_per_second: 2.416
11
  * train_loss: 0.3722160959110207
12
+ * epoch: 5.0
13
+
14
+ ----
15
+ ### How to use
16
+ ```
17
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
18
+
19
+ tokenizer = AutoTokenizer.from_pretrained("seongju/klue-tc-bert-base-multilingual-cased")
20
+
21
+ model = AutoModelForSequenceClassification.from_pretrained("seongju/klue-tc-bert-base-multilingual-cased")
22
+ mapping = {0: 'IT๊ณผํ•™', 1: '๊ฒฝ์ œ', 2: '์‚ฌํšŒ', 3: '์ƒํ™œ๋ฌธํ™”', 4: '์„ธ๊ณ„', 5: '์Šคํฌ์ธ ', 6: '์ •์น˜'}
23
+ inputs = tokenizer("๋ฐฑ์‹  ํšŒํ”ผ ๊ฐ€๋Šฅ์„ฑ? ๋‚จ๋ฏธ์—์„œ ์ƒˆ๋กœ์šด ๋ณ€์ด ๋ฐ”์ด๋Ÿฌ์Šค ๊ธ‰์† ํ™•์‚ฐ ", padding=True, truncation=True, max_length=128, return_tensors="pt")
24
+ outputs = model(**inputs)
25
+ probs = outputs[0].softmax(1)
26
+ output = mapping[probs.argmax().item()]
27
+ ```