Downloads · 30 days
6
2% of all-time downloads
stress-roberta-base
Amalq/stress-roberta-base
stress-roberta-base is a fill-mask model from Amalq. Use it when you need the model to fill a missing word. It is set up for transformers. The card lists the license as apache-2.0.
This model was initialized with roberta-base and trained with Stress-SMHD, which is a subset of Self-Reported Mental Health Diagnoses (SMHD) dataset which consists of Reddit posts by patients with anxiety, PTSD (Post…
All-time downloads
342
Public
Repo size
998 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin499 MB · 99%
At a glance
- Task
- Fill-Mask
- Library
- transformers
- License
- apache-2.0
- Model type
- roberta
- Access
- Public
- Created
- Apr 27, 2023
- Updated
- Apr 28, 2023
- SHA
- 59da879e
Try a prompt
Datasets
Model card
From the Hugging Face model README
StressRoberta model:
This model was initialized with roberta-base and trained with Stress-SMHD, which is a subset of Self-Reported Mental Health Diagnoses (SMHD) dataset which consists of Reddit posts by patients with anxiety, PTSD (Post Traumatic Stress Disorder), or depression. We follow the standard pretraining protocols of RoBERTa with Huggingface’s Transformers library.
Usage Load the model via Huggingface’s Transformers library:
from transformers import AutoTokenizer,
AutoModel tokenizer = AutoTokenizer.from_pretrained("Amalq/stress-roberta-base")
model = AutoModel.from_pretrained("Amalq/stress-roberta-base")
Perplexity of this model is: 5.22
- Task
- Fill-Mask
- Library
- transformers
- Type
- roberta
- License
- apache-2.0
- Languages
- en
- Created
- Apr 27, 2023
- Updated
- Apr 28, 2023