Update README.md
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ metrics:
|
|
| 15 |
- accuracy
|
| 16 |
- f1
|
| 17 |
model-index:
|
| 18 |
-
- name: roberta-depression-severity
|
| 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
|
| 33 |
|
| 34 |
## Model Description
|
| 35 |
|
| 36 |
-
This is a **RoBERTa-base** model fine-tuned
|
| 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
|