Downloads · 30 days
6
6% of all-time downloads
Twna-Jane16/Neurochat_MentalBERT_depression
Neurochat_MentalBERT_depression is a machine learning model from Twna-Jane16. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This MentalBERT-based model is fine-tuned to detect signs of depression from text input. It leverages the pretrained mental/mental-bert-base-uncased model and has been trained on multiple mental health datasets. It ca…
Downloads · 30 days
6
6% of all-time downloads
All-time downloads
100
Public
Parameters
109M
876 MB on disk
Likes
1
Public
Click a slice to open those files.
.bin438 MB · 50%
From the Hugging Face model README
This MentalBERT-based model is fine-tuned to detect signs of depression from text input.
It leverages the pretrained mental/mental-bert-base-uncased model and has been trained on multiple mental health datasets.
It can assist in early screening of depressive symptoms through conversational AI applications.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Twna-Jane16/Neurochat_MentalBERT_depression")
model = AutoModelForSequenceClassification.from_pretrained("Twna-Jane16/Neurochat_MentalBERT_depression")