webesama commited on
Commit
ec313ee
·
verified ·
1 Parent(s): 764ae2b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -14
README.md CHANGED
@@ -24,21 +24,18 @@ tags:
24
 
25
  This model was developed to support standardized, scalable mental health assessments in both clinical and low-resource settings.
26
 
27
- ---
28
-
29
 
30
- ## 🧠 Model Details
31
 
32
  - **Base model**: `bert-base-german-cased`
33
- - **Task**: Ordinal regression/classification (scores 0–6)
34
- - **Language**: German 🇩🇪
35
  - **Input**: Text (dialogue segment grouped by MADRS topic)
36
  - **Output**: Predicted score for each MADRS item (rounded integer 0–6)
37
  - **Training data**: Mix of real and synthetic clinician–patient interviews (MADRS-structured)
38
 
39
- ---
40
 
41
- ## 💡 Intended Use
42
 
43
  This model is intended for research and development use. It is not a certified medical device. The goal is to:
44
  - Explore AI-assisted symptom severity assessment
@@ -88,7 +85,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
88
  model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
89
  ```
90
 
91
- ### 📝 Predict on a full structured interview / Run inference:
92
  Assume you have a conversation log like this:
93
 
94
  ```python
@@ -113,20 +110,17 @@ print(scores)
113
 
114
  ---
115
 
116
- ## 🙏 Acknowledgements
117
 
118
  Model trained and released by [Samantha Weber](https://github.com/webersamantha). Research conducted as part of efforts to improve AI-driven mental health tools. Thanks to all clinicians and collaborators who contributed to the annotated MADRS dataset.
119
 
120
- ---
121
-
122
 
123
- ## 🧪 Evaluation
124
 
125
  The model was evaluated on a held-out clinical validation set and achieved strong performance under both strict and flexible scoring criteria (±1 deviation tolerance). See publication for full metrics.
126
 
127
- ---
128
 
129
- ## 🧪 Citation
130
 
131
  If you use this model, please cite:
132
  > Weber, S. et al. (2025). "Using a Fine-tuned Large Language Model for Symptom-based Depression Evaluation" *Preprint*. https://doi.org/10.21203/rs.3.rs-6555767/v1
 
24
 
25
  This model was developed to support standardized, scalable mental health assessments in both clinical and low-resource settings.
26
 
 
 
27
 
28
+ ## Model Details
29
 
30
  - **Base model**: `bert-base-german-cased`
31
+ - **Task**: Ordinal regression (scores 0–6)
32
+ - **Language**: German
33
  - **Input**: Text (dialogue segment grouped by MADRS topic)
34
  - **Output**: Predicted score for each MADRS item (rounded integer 0–6)
35
  - **Training data**: Mix of real and synthetic clinician–patient interviews (MADRS-structured)
36
 
 
37
 
38
+ ## Intended Use
39
 
40
  This model is intended for research and development use. It is not a certified medical device. The goal is to:
41
  - Explore AI-assisted symptom severity assessment
 
85
  model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
86
  ```
87
 
88
+ ### Predict on a full structured interview / Run inference:
89
  Assume you have a conversation log like this:
90
 
91
  ```python
 
110
 
111
  ---
112
 
113
+ ## Acknowledgements
114
 
115
  Model trained and released by [Samantha Weber](https://github.com/webersamantha). Research conducted as part of efforts to improve AI-driven mental health tools. Thanks to all clinicians and collaborators who contributed to the annotated MADRS dataset.
116
 
 
 
117
 
118
+ ## Evaluation
119
 
120
  The model was evaluated on a held-out clinical validation set and achieved strong performance under both strict and flexible scoring criteria (±1 deviation tolerance). See publication for full metrics.
121
 
 
122
 
123
+ ## Citation
124
 
125
  If you use this model, please cite:
126
  > Weber, S. et al. (2025). "Using a Fine-tuned Large Language Model for Symptom-based Depression Evaluation" *Preprint*. https://doi.org/10.21203/rs.3.rs-6555767/v1