Instructions to use Sharpaxis/Mental-Health-RoBERTa with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Sharpaxis/Mental-Health-RoBERTa with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Sharpaxis/Mental-Health-RoBERTa")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Sharpaxis/Mental-Health-RoBERTa") model = AutoModelForSequenceClassification.from_pretrained("Sharpaxis/Mental-Health-RoBERTa") - Notebooks
- Google Colab
- Kaggle
Data Labels Explanation
What are the classification schema from label 0 to 6????
Sorry, I forgot to mention in model card, here you go :
'Normal':0,'Depression':1,'Suicidal':2,'Anxiety':3,'Bipolar':4,'Stress':5,'Personality disorder':6
Sorry, I forgot to mention in model card, here you go :
'Normal':0,'Depression':1,'Suicidal':2,'Anxiety':3,'Bipolar':4,'Stress':5,'Personality disorder':6
Thank you so much
Sorry, I forgot to mention in model card, here you go :
'Normal':0,'Depression':1,'Suicidal':2,'Anxiety':3,'Bipolar':4,'Stress':5,'Personality disorder':6
Just curious, do you have any available publication with the model? or do you have more specific information related the dataset you've used? Thanks.
https://www.kaggle.com/datasets/suchintikasarkar/sentiment-analysis-for-mental-health here is the used dataset