LimYeri commited on
Commit
c55edaf
ยท
verified ยท
1 Parent(s): 9167307

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -7,7 +7,7 @@ tags: []
7
 
8
  ## Model Description
9
  KoELECTRA ๊ธฐ๋ฐ˜์˜ ํ•œ๊ตญ์–ด(ํŠนํžˆ ์ผ๊ธฐ/์‹ฌ๋ฆฌ ๊ธฐ๋ก) ๊ฐ์ • ๋ถ„๋ฅ˜ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.<br>
10
- ํ…์ŠคํŠธ์—์„œ 8๊ฐ€์ง€ ๊ฐ์ •(๊ธฐ์จ, ์„ค๋ ˜, ํ‰๋ฒ”ํ•จ, ๋†€๋ผ์›€, ๋ถˆ์พŒํ•จ, ๋‘๋ ค์›€, ์Šฌํ””, ๋ถ„๋…ธ)์„ ์ž๋™์œผ๋กœ ์ธ์‹ํ•ฉ๋‹ˆ๋‹ค.
11
 
12
  - **Model type:** Text Classification (Emotion Recognition)
13
  - **Language (NLP):** Korean (ํ•œ๊ตญ์–ด, ko)
@@ -15,7 +15,7 @@ KoELECTRA ๊ธฐ๋ฐ˜์˜ ํ•œ๊ตญ์–ด(ํŠนํžˆ ์ผ๊ธฐ/์‹ฌ๋ฆฌ ๊ธฐ๋ก) ๊ฐ์ • ๋ถ„๋ฅ˜ ๋ชจ๋ธ
15
  - **Finetuned from model:** monologg/koelectra-base-v3-discriminator
16
 
17
  ## Emotion Classes
18
- ์ด ๋ชจ๋ธ์€ ์ž…๋ ฅ๋œ ํ•œ๊ตญ์–ด ๋ฌธ์žฅ ๋˜๋Š” ์ผ๊ธฐ ํ…์ŠคํŠธ์˜ ์ฃผ์š” ๊ฐ์ •์„ ์•„๋ž˜ 8๊ฐœ ํด๋ž˜์Šค ์ค‘ ํ•˜๋‚˜๋กœ ๋ถ„๋ฅ˜ํ•ฉ๋‹ˆ๋‹ค.
19
  | Emotion (๊ฐ์ •) |
20
  |----------------|
21
  | ๊ธฐ์จ |
@@ -48,7 +48,7 @@ def predict(text):
48
  label_id = probs.argmax().item()
49
  return label_id, probs.tolist()
50
 
51
- print(predict("์˜ค๋Š˜ ํ•˜๋ฃจ๊ฐ€ ์œ ๋‚œํžˆ ํž˜๋“ค๊ณ  ์ง€์ณค๋‹ค."))
52
  ```
53
 
54
  ---
 
7
 
8
  ## Model Description
9
  KoELECTRA ๊ธฐ๋ฐ˜์˜ ํ•œ๊ตญ์–ด(ํŠนํžˆ ์ผ๊ธฐ/์‹ฌ๋ฆฌ ๊ธฐ๋ก) ๊ฐ์ • ๋ถ„๋ฅ˜ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.<br>
10
+ ํ…์ŠคํŠธ์—์„œ 8๊ฐ€์ง€ ๊ฐ์ •(๊ธฐ์จ, ์„ค๋ ˜, ํ‰๋ฒ”ํ•จ, ๋†€๋ผ์›€, ๋ถˆ์พŒํ•จ, ๋‘๋ ค์›€, ์Šฌํ””, ๋ถ„๋…ธ)์„ ์ธ์‹ํ•ฉ๋‹ˆ๋‹ค.
11
 
12
  - **Model type:** Text Classification (Emotion Recognition)
13
  - **Language (NLP):** Korean (ํ•œ๊ตญ์–ด, ko)
 
15
  - **Finetuned from model:** monologg/koelectra-base-v3-discriminator
16
 
17
  ## Emotion Classes
18
+ ์ด ๋ชจ๋ธ์€ ์ž…๋ ฅ๋œ ํ•œ๊ตญ์–ด ๋ฌธ์žฅ์˜ ์ฃผ์š” ๊ฐ์ •์„ ์•„๋ž˜ 8๊ฐœ ํด๋ž˜์Šค ์ค‘ ํ•˜๋‚˜๋กœ ๋ถ„๋ฅ˜ํ•ฉ๋‹ˆ๋‹ค.
19
  | Emotion (๊ฐ์ •) |
20
  |----------------|
21
  | ๊ธฐ์จ |
 
48
  label_id = probs.argmax().item()
49
  return label_id, probs.tolist()
50
 
51
+ print(predict("์˜ค๋Š˜ ์ •๋ง ๊ธฐ๋ถ„์ด ์ข‹๊ณ  ํ–‰๋ณตํ•œ ํ•˜๋ฃจ์˜€์–ด!"))
52
  ```
53
 
54
  ---