{ "_name_or_path": "mental/mental-roberta-base", "architectures": [ "RobertaForSequenceClassification" ], "attention_probs_dropout_prob": 0.1, "bos_token_id": 0, "classifier_dropout": null, "eos_token_id": 2, "gradient_checkpointing": false, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { "0": "Mental filter", "1": "Mind Reading", "2": "No Distortion", "3": "Labeling", "4": "Emotional Reasoning", "5": "Should statements", "6": "Magnification", "7": "All-or-nothing thinking", "8": "Overgeneralization", "9": "Personalization", "10": "Fortune-telling" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "All-or-nothing thinking": 7, "Emotional Reasoning": 4, "Fortune-telling": 10, "Labeling": 3, "Magnification": 6, "Mental filter": 0, "Mind Reading": 1, "No Distortion": 2, "Overgeneralization": 8, "Personalization": 9, "Should statements": 5 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514, "model_type": "roberta", "num_attention_heads": 12, "num_hidden_layers": 12, "pad_token_id": 1, "position_embedding_type": "absolute", "problem_type": "single_label_classification", "torch_dtype": "float32", "transformers_version": "4.38.0", "type_vocab_size": 1, "use_cache": true, "vocab_size": 50265 }