Akashpaul123 commited on
Commit
8c7724e
·
verified ·
1 Parent(s): 7869306

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -6
README.md CHANGED
@@ -15,7 +15,7 @@ metrics:
15
  - accuracy
16
  - f1
17
  model-index:
18
- - name: roberta-depression-severity-lora
19
  results:
20
  - task:
21
  type: text-classification
@@ -29,11 +29,11 @@ model-index:
29
  name: F1 Score
30
  ---
31
 
32
- # RoBERTa Depression Severity Classifier (LoRA)
33
 
34
  ## Model Description
35
 
36
- This is a **RoBERTa-base** model fine-tuned with **LoRA (Low-Rank Adaptation)** for depression severity classification into 4 categories:
37
  - **Minimum** (BDI-3: 0-9)
38
  - **Mild** (BDI-3: 10-18)
39
  - **Moderate** (BDI-3: 19-29)
@@ -41,11 +41,9 @@ This is a **RoBERTa-base** model fine-tuned with **LoRA (Low-Rank Adaptation)**
41
 
42
  ## Model Details
43
 
44
- - **Base Model**: `roberta-base`
45
- - **Method**: LoRA (Parameter-Efficient Fine-Tuning)
46
  - **Total Parameters**: 124,648,708
47
  - **Trainable Parameters**: 1,183,492 (0.95%)
48
- - **LoRA Rank**: 16
49
  - **Model Size**: ~10-20MB (adapters only)
50
 
51
  ## Performance
 
15
  - accuracy
16
  - f1
17
  model-index:
18
+ - name: roberta-depression-severity
19
  results:
20
  - task:
21
  type: text-classification
 
29
  name: F1 Score
30
  ---
31
 
32
+ # RoBERTa Depression Severity Classifier
33
 
34
  ## Model Description
35
 
36
+ This is a **RoBERTa-base** model fine-tuned for depression severity classification into 4 categories:
37
  - **Minimum** (BDI-3: 0-9)
38
  - **Mild** (BDI-3: 10-18)
39
  - **Moderate** (BDI-3: 19-29)
 
41
 
42
  ## Model Details
43
 
44
+ - **Base Model**: `roberta-base'
 
45
  - **Total Parameters**: 124,648,708
46
  - **Trainable Parameters**: 1,183,492 (0.95%)
 
47
  - **Model Size**: ~10-20MB (adapters only)
48
 
49
  ## Performance