| # bert-large-depression-classification-model | |
| To access this model, just copy & paste the following code: | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline('text-classification', model='ardavey/bert-large-depression-classification-model') | |
| ``` | |
| **Example usage**: | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline('text-classification', model='ardavey/bert-large-depression-classification-model') | |
| text_samples = [ | |
| "The things I used to love don’t bring me any joy anymore. Everything feels empty", | |
| "I often wonder if things would be better if I just disappeared.", | |
| "Sometimes things get tough, but I know I have the strength to get through it." | |
| ] | |
| predictions = classifier(text_samples) | |
| for sample, prediction in zip(text_samples, predictions): | |
| print(f"Text: {sample}") | |
| print(f"Prediction: {'Depressed' if prediction['label'] == 'LABEL_1' else 'Not Depressed'}, Score: {prediction['score']}") | |
| print() | |
| ``` | |
| **Output**: | |
| ``` | |
| Text: The things I used to love don’t bring me any joy anymore. Everything feels empty | |
| Prediction: Depressed, Score: 0.9950354099273682 | |
| Text: I often wonder if things would be better if I just disappeared. | |
| Prediction: Depressed, Score: 0.970693051815033 | |
| Text: Sometimes things get tough, but I know I have the strength to get through it. | |
| Prediction: Not Depressed, Score: 0.9896683692932129 | |
| ``` | |