Instructions to use Amalq/stress-roberta-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Amalq/stress-roberta-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="Amalq/stress-roberta-base")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("Amalq/stress-roberta-base") model = AutoModelForMaskedLM.from_pretrained("Amalq/stress-roberta-base") - Notebooks
- Google Colab
- Kaggle
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
- Downloads last month
- 39