Cyplus72 commited on
Commit
1b88b40
·
verified ·
1 Parent(s): 9af985a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +38 -1
README.md CHANGED
@@ -12,12 +12,49 @@ 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
  ## 사용 방법
36
  ```python
37
+ from peft import PeftModel
38
+
39
+ # 베이스 모델 로드 (분류용)
40
+ print("베이스 모델 로딩")
41
+ base_model_reload = AutoModelForSequenceClassification.from_pretrained(
42
+ "klue/bert-base",
43
+ num_labels=2
44
+ )
45
+
46
+ # 업로드한 LoRA 어댑터 로드
47
+ print(f"LoRA 어댑터 로딩: Cyplus72/nsmc-sentiment-lora")
48
+ model_reload = PeftModel.from_pretrained(base_model_reload, model_name_upload)
49
+ tokenizer_reload = AutoTokenizer.from_pretrained(model_name_upload)
50
+
51
+ # GPU로 이동
52
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
53
+ model_reload = model_reload.to(device)
54
+ model_reload.eval()
55
 
56
+ print("모델 로드 완료!")
57
+ print("이제 이 코드로 어디서든 내 모델을 사용할 수 있습니다!")
58
  ```
59
 
60
  **Note**: 이 모델은 교육 목적으로 만들어졌습니다.