Update README.md
Browse files
README.md
CHANGED
|
@@ -44,18 +44,25 @@ It achieves the following results on the evaluation set:
|
|
| 44 |
|
| 45 |
## Model description
|
| 46 |
|
| 47 |
-
|
|
|
|
| 48 |
|
| 49 |
## Intended uses & limitations
|
| 50 |
|
| 51 |
-
|
|
|
|
| 52 |
|
| 53 |
## Training and evaluation data
|
| 54 |
|
| 55 |
-
|
|
|
|
|
|
|
| 56 |
|
| 57 |
## Training procedure
|
| 58 |
|
|
|
|
|
|
|
|
|
|
| 59 |
### Training hyperparameters
|
| 60 |
|
| 61 |
The following hyperparameters were used during training:
|
|
|
|
| 44 |
|
| 45 |
## Model description
|
| 46 |
|
| 47 |
+
This model is a transformer-based model that has been fine-tuned on a dataset of Reddit posts related to depression.
|
| 48 |
+
The model can be used to classify posts as either depression or not depression.
|
| 49 |
|
| 50 |
## Intended uses & limitations
|
| 51 |
|
| 52 |
+
This model is intended to be used for research purposes. It is not yet ready for production use.
|
| 53 |
+
The model has been trained on a dataset of English-language posts, so it may not be accurate for other languages.
|
| 54 |
|
| 55 |
## Training and evaluation data
|
| 56 |
|
| 57 |
+
The model was trained on the mrjunos/depression-reddit-cleaned dataset, which contains approximately 7,000 labeled instances.
|
| 58 |
+
The data was split into Train and Test using train_test_split(test_size=0.2, seed=42)
|
| 59 |
+
The dataset consists of two main features: 'text' and 'label'. The 'text' feature contains the text data from Reddit posts related to depression, while the 'label' feature indicates whether a post is classified as depression or not.
|
| 60 |
|
| 61 |
## Training procedure
|
| 62 |
|
| 63 |
+
You can find here the steps I followed to train this model:
|
| 64 |
+
https://github.com/mrjunos/machine_learning/blob/main/NLP-fine_tunning-hugging_face_model.ipynb
|
| 65 |
+
|
| 66 |
### Training hyperparameters
|
| 67 |
|
| 68 |
The following hyperparameters were used during training:
|