{ "_name_or_path": "michiyasunaga/BioLinkBERT-base", "architectures": [ "BertForSequenceClassification" ], "attention_probs_dropout_prob": 0.1, "classifier_dropout": null, "gradient_checkpointing": false, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { "0": "Epidemic Forecasting", "1": "Treatment", "2": "Prevention", "3": "Mechanism", "4": "Case Report", "5": "Transmission", "6": "Diagnosis", "7": "Long Covid", "8": "General Info", "9": "Communication", "10": "Contact Tracing", "11": "Drug Targets", "12": "Education", "13": "Effect on Medical Specialties", "14": "Health Policy", "15": "Healthcare Workers", "16": "Imaging", "17": "Immunology", "18": "Inequality", "19": "Medical Devices", "20": "Misinformation", "21": "Model Systems & Tools", "22": "Molecular Biology", "23": "Non-human", "24": "Non-medical", "25": "Pediatrics", "26": "Prevalence", "27": "Psychology", "28": "Recommendations", "29": "Risk Factors", "30": "Surveillance", "31": "Vaccines", "32": "AI/ML/statistics", "33": "Comorbidities", "34": "Telemedicine", "35": "Drug Design and Discovery" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "AI/ML/statistics": 32, "Case Report": 4, "Communication": 9, "Comorbidities": 33, "Contact Tracing": 10, "Diagnosis": 6, "Drug Design and Discovery": 35, "Drug Targets": 11, "Education": 12, "Effect on Medical Specialties": 13, "Epidemic Forecasting": 0, "General Info": 8, "Health Policy": 14, "Healthcare Workers": 15, "Imaging": 16, "Immunology": 17, "Inequality": 18, "Long Covid": 7, "Mechanism": 3, "Medical Devices": 19, "Misinformation": 20, "Model Systems & Tools": 21, "Molecular Biology": 22, "Non-human": 23, "Non-medical": 24, "Pediatrics": 25, "Prevalence": 26, "Prevention": 2, "Psychology": 27, "Recommendations": 28, "Risk Factors": 29, "Surveillance": 30, "Telemedicine": 34, "Transmission": 5, "Treatment": 1, "Vaccines": 31 }, "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.28.0", "type_vocab_size": 2, "use_cache": true, "vocab_size": 28895 }