Downloads · 30 days
9
0% of all-time downloads
psychbert-finetuned-multiclass
mnaylor/psychbert-finetuned-multiclass
psychbert-finetuned-multiclass is a text classification model from mnaylor. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This is a version of https://huggingface.co/mnaylor/psychbert-cased which was fine-tuned to illustrate performance on a multi-class classification problem involving the detection of different types of language relatin…
All-time downloads
3.1K
Public
Repo size
867 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin433 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Mar 29, 2024
- Updated
- Mar 29, 2024
- SHA
- e5493449
Try a prompt
Model card
From the Hugging Face model README
PsychBERT Fine-Tuned on Simple Multi-class Problem
This is a version of https://huggingface.co/mnaylor/psychbert-cased which was fine-tuned to illustrate performance on a multi-class classification problem involving the detection of different types of language relating to mental health. The classes are as follows:
- 0: Negative / unrelated to mental health
- 1: Mental illnesses
- 2: Anxiety
- 3: Depression
- 4: Social anxiety
- 5: Loneliness
The dataset for this model was taken from Reddit and Twitter, and labels were assigned based on the post appearing in certain subreddits or containing certain hashtags. For more information, see the PsychBERT paper.
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Created
- Mar 29, 2024
- Updated
- Mar 29, 2024