Instructions to use mental/mental-bert-base-uncased with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mental/mental-bert-base-uncased with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="mental/mental-bert-base-uncased")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("mental/mental-bert-base-uncased") model = AutoModelForMaskedLM.from_pretrained("mental/mental-bert-base-uncased") - Notebooks
- Google Colab
- Kaggle
Labels description
#4
by nikilas - opened
We get output labels 0 to 5; where can I find the meaning of each label; I suppose it is mapped to no mental disorder, depression, suicide, stress, anxiety, and bipolar. Where can I find the mapping?
We get output labels 0 to 5; where can I find the meaning of each label; I suppose it is mapped to no mental disorder, depression, suicide, stress, anxiety, and bipolar. Where can I find the mapping?
I have exactly the same question, do you figure it out?