How to use mahir2021/mental-bert-depEMO with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahir2021/mental-bert-depEMO")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mahir2021/mental-bert-depEMO") model = AutoModelForSequenceClassification.from_pretrained("mahir2021/mental-bert-depEMO", device_map="auto")
c71786f 1acf416 c71786f 1acf416 c71786f 1acf416 c71786f
1
2
3
4
5
6
7
8
9
10
11
{ "seed": 42, "max_len": 256, "hidden_dim": 256, "dropout": 0.3, "batch_train": 16, "batch_eval": 32, "epochs": 5, "lr": 2e-05, "weight_decay": 0.01 }
