Instructions to use mnaylor/psychbert-finetuned-mentalhealth with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mnaylor/psychbert-finetuned-mentalhealth with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mnaylor/psychbert-finetuned-mentalhealth")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mnaylor/psychbert-finetuned-mentalhealth") model = AutoModelForSequenceClassification.from_pretrained("mnaylor/psychbert-finetuned-mentalhealth") - Notebooks
- Google Colab
- Kaggle
PsychBERT Fine-tuned to Identify Mental Health Conversations
This is a version of https://huggingface.co/mnaylor/psychbert-cased which was fine-tuned to identify text relevant to mental health discussions (e.g. depression and
anxiety). It can be loaded with the standard AutoModelFor* and AutoTokenizer classes, and its outputs are logits corresponding to mental health topics (index 1)
and all other text (index 0).
- Downloads last month
- 7