{ "_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract", "architectures": [ "BertForSequenceClassification" ], "attention_probs_dropout_prob": 0.1, "classifier_dropout": null, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { "0": "Psychiatric condition", "1": "Depression", "10": "(Chronic) Pain", "11": "Palliative Setting", "12": "Recreational Drug Use", "13": "Healthy Participants", "2": "Anxiety", "3": "Post-traumatic stress disorder (PTSD)", "4": "Alcoholism", "5": "Other addictions (e.g. smoking)", "6": "Anorexia", "7": "Alzheimer\u2019s disease", "8": "Non-Alzheimer dementia", "9": "Substance abuse" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "(Chronic) Pain": "10", "Alcoholism": "4", "Alzheimer\u2019s disease": "7", "Anorexia": "6", "Anxiety": "2", "Depression": "1", "Healthy Participants": "13", "Non-Alzheimer dementia": "8", "Other addictions (e.g. smoking)": "5", "Palliative Setting": "11", "Post-traumatic stress disorder (PTSD)": "3", "Psychiatric condition": "0", "Recreational Drug Use": "12", "Substance abuse": "9" }, "layer_norm_eps": 1e-12, "max_position_embeddings": 512, "model_type": "bert", "num_attention_heads": 12, "num_hidden_layers": 12, "pad_token_id": 0, "position_embedding_type": "absolute", "problem_type": "multi_label_classification", "torch_dtype": "float32", "transformers_version": "4.49.0", "type_vocab_size": 2, "use_cache": true, "vocab_size": 30522 }