yooranyeong commited on
Commit
eb542f8
ยท
verified ยท
1 Parent(s): 3eabeff

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +38 -0
README.md CHANGED
@@ -12,12 +12,50 @@ 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
+ - ** ใ…‡
33
 
34
  ## ๋ชจ๋ธ ์ •๋ณด
35
 
36
  ## ์‚ฌ์šฉ ๋ฐฉ๋ฒ•
37
  ```python
38
+ from peft import PeftModel
39
+
40
+ # ๋ฒ ์ด์Šค ๋ชจ๋ธ ๋กœ๋“œ (๋ถ„๋ฅ˜์šฉ)
41
+ print("๋ฒ ์ด์Šค ๋ชจ๋ธ ๋กœ๋”ฉ")
42
+ base_model_reload = AutoModelForSequenceClassification.from_pretrained(
43
+ "klue/bert-base",
44
+ num_labels=2
45
+ )
46
+
47
+ # ์—…๋กœ๋“œํ•œ LoRA ์–ด๋Œ‘ํ„ฐ ๋กœ๋“œ
48
+ print(f"LoRA ์–ด๋Œ‘ํ„ฐ ๋กœ๋”ฉ: yooranyeong/nsmc-sentiment-lora")
49
+ model_reload = PeftModel.from_pretrained(base_model_reload, model_name_upload)
50
+ tokenizer_reload = AutoTokenizer.from_pretrained(model_name_upload)
51
+
52
+ # GPU๋กœ ์ด๋™
53
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
54
+ model_reload = model_reload.to(device)
55
+ model_reload.eval()
56
 
57
+ print("๋ชจ๋ธ ๋กœ๋“œ ์™„๋ฃŒ!")
58
+ print("์ด์ œ ์ด ์ฝ”๋“œ๋กœ ์–ด๋””์„œ๋“  ๋‚ด ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค!")
59
  ```
60
 
61
  **Note**: ์ด ๋ชจ๋ธ์€ ๊ต์œก ๋ชฉ์ ์œผ๋กœ ๋งŒ๋“ค์–ด์กŒ์Šต๋‹ˆ๋‹ค.