Accuracy
TP | TN | |
---|---|---|
FP | 492 | 62 |
FN | 39 | 467 |
์ ํ๋: 89.9%(492+267/1000)
์ ํ๋๊ฐ ๊ฑฐ์ 90%์ ๊ทผ์ ํฉ๋๋ค.
-Train ๋ฐ์ดํฐ๋ nsmc['train']์ shuffleํ ๋ค์์ ์์ 2000๊ฐ๋ฅผ ํ์ฉํ์์ต๋๋ค
-Test ๋ฐ์ดํฐ๋ nsmc['train']์ shuffleํ ๋ค์์ ์์ 1000๊ฐ๋ฅผ ํ์ฉํ์์ต๋๋ค.
-1000๊ฐ์ train ๋ฐ์ดํฐ ์ค 890๊ฐ๋ฅผ ์ ํํ๊ฒ ๋ถ๋ฅํ์์ต๋๋ค.
-Base Model์ KT-AI/midm-bitext-S-7B-inst-v1๋ก base model์ Lora ์ด๋ํฐ๋ฅผ ๋ถ์ฌ์ SFTtrainer๋ฅผ ํตํ์ฌ ์๋ก์ด ๋ฐ์ดํฐ์ nsmc์ ๋ํด์ ๋ฏธ์ธํ๋์ ํ์์ต๋๋ค.
-๋ฏธ์ธํ๋ํ ๋ก๋ผ ์ด๋ํฐ๋ฅผ ํ๊น ํ์ด์ค์ ์ ๋ก๋ํ ํ 4๋นํธ๋ก ์์ํ๋ base model์ ๋ถ์ธ ํ ์ถ๋ก ์ ํ์์ต๋๋ค.
-์์ 2000๊ฐ์ train_dataset์ ๋ํด์ ํ๋ จ์ ํ ํ ์์ 1000๊ฐ์ test_dataset์ ๋ํด์ ์ถ๋ก ์ ํ ๊ฒฐ๊ณผ๋ ์์ ํ์ ๊ฐ์ต๋๋ค.
-์ ํ๋ ํฅ์์ ์ํ์ฌ test ๋ฐ์ดํฐ์ ์ ํฌ๊ธฐ๋ฅผ ๋๋ฆฌ๋ ค ํ์์ผ๋, GPU ์ฉ๋ ์ ํ์ผ๋ก ์ธํ์ฌ ์ง์์ ์ผ๋ก ์ค๋ฅ๊ฐ ๋ฐ์ํ์์ต๋๋ค.
-์ ํ๋๋ฅผ ์ฌ๋ฆฌ๋ ๋ฐฉ๋ฒ: seq_length๋ฅผ ํค์ด๋ค. 512 ์ ๋๋ก
- Downloads last month
- 3
Model tree for kayla0913/hw-midm-7B-nsmc
Base model
KT-AI/midm-bitext-S-7B-inst-v1