jhice commited on
Commit
7d7f35d
ยท
verified ยท
1 Parent(s): e446500

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +37 -1
README.md CHANGED
@@ -12,12 +12,48 @@ language:
12
  datasets:
13
  - nsmc
14
  ---
 
 
15
 
16
- ## ๋ชจ๋ธ ์ •๋ณด
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
  ## ์‚ฌ์šฉ ๋ฐฉ๋ฒ•
19
  ```python
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
 
 
21
  ```
22
 
23
  **Note**: ์ด ๋ชจ๋ธ์€ ๊ต์œก ๋ชฉ์ ์œผ๋กœ ๋งŒ๋“ค์–ด์กŒ์Šต๋‹ˆ๋‹ค.
 
12
  datasets:
13
  - nsmc
14
  ---
15
+ # NSMC ๊ฐ์ • ๋ถ„์„ LoRA ๋ชจ๋ธ
16
+ NSMC ๋ฐ์ดํ„ฐ์…‹์œผ๋กœ ํŒŒ์ธํŠœ๋‹๋œ ํ•œ๊ตญ์–ด ๊ฐ์ • ๋ถ„์„ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.
17
 
18
+ ## ๋ชจ๋ธ ์„ค๋ช…
19
+ - **๋ฒ ์ด์Šค ๋ชจ๋ธ**: klue/bert_base
20
+ - **ํŒŒ์ธ ํŠœ๋‹ ๋ฐฉ๋ฒ•**: LoRA
21
+ - **์–ธ์–ด**: ํ•œ๊ตญ์–ด
22
+
23
+ ## ์„ฑ๋Šฅ
24
+ - **์ตœ์ข… ์„ฑ๋Šฅ**: 85%
25
+
26
+ ## ํ•™์Šต์ •๋ณด
27
+ ### ๋ฐ์ดํ„ฐ์…‹
28
+ -**์ด๋ฆ„**: NSMC
29
+ -**ํ•™์Šต ๋ฐ์ดํ„ฐ**: 10000
30
+
31
+ ## ํ•™์Šต ์„ค์ •
32
+ -**์—ํญ**:3
33
 
34
  ## ์‚ฌ์šฉ ๋ฐฉ๋ฒ•
35
  ```python
36
+ from peft import PeftModel
37
+
38
+ # ๋ฒ ์ด์Šค ๋ชจ๋ธ ๋กœ๋“œ (๋ถ„๋ฅ˜์šฉ)
39
+ print("๋ฒ ์ด์Šค ๋ชจ๋ธ ๋กœ๋”ฉ")
40
+ base_model_reload = AutoModelForSequenceClassification.from_pretrained(
41
+ "klue/bert-base",
42
+ num_labels=2
43
+ )
44
+
45
+ # ์—…๋กœ๋“œํ•œ LoRA ์–ด๋Œ‘ํ„ฐ ๋กœ๋“œ
46
+ print(f"LoRA ์–ด๋Œ‘ํ„ฐ ๋กœ๋”ฉ: jhice/nsmc-sentiment-lora")
47
+ model_reload = PeftModel.from_pretrained(base_model_reload, model_name_upload)
48
+ tokenizer_reload = AutoTokenizer.from_pretrained(model_name_upload)
49
+
50
+ # GPU๋กœ ์ด๋™
51
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
52
+ model_reload = model_reload.to(device)
53
+ model_reload.eval()
54
 
55
+ print("๋ชจ๋ธ ๋กœ๋“œ ์™„๋ฃŒ!")
56
+ print("์ด์ œ ์ด ์ฝ”๋“œ๋กœ ์–ด๋””์„œ๋“  ๋‚ด ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค!")
57
  ```
58
 
59
  **Note**: ์ด ๋ชจ๋ธ์€ ๊ต์œก ๋ชฉ์ ์œผ๋กœ ๋งŒ๋“ค์–ด์กŒ์Šต๋‹ˆ๋‹ค.