config.json

{
  "_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": "Clinical Trial",
    "10": "Communication",
    "11": "Contact Tracing",
    "12": "Drug Targets",
    "13": "Education",
    "14": "Effect on Medical Specialties",
    "15": "Health Policy",
    "16": "Healthcare Workers",
    "17": "Imaging",
    "18": "Immunology",
    "19": "Inequality",
    "20": "Medical Devices",
    "21": "Misinformation",
    "22": "Model Systems & Tools",
    "23": "Molecular Biology",
    "24": "Non-human",
    "25": "Non-medical",
    "26": "Pediatrics",
    "27": "Prevalence",
    "28": "Psychology",
    "29": "Recommendations",
    "30": "Risk Factors",
    "31": "Surveillance",
    "32": "Vaccines"
  },
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "label2id": {
    "Case Report": 4,
    "Clinical Trial": 9,
    "Communication": 10,
    "Contact Tracing": 11,
    "Diagnosis": 6,
    "Drug Targets": 12,
    "Education": 13,
    "Effect on Medical Specialties": 14,
    "Epidemic Forecasting": 0,
    "General Info": 8,
    "Health Policy": 15,
    "Healthcare Workers": 16,
    "Imaging": 17,
    "Immunology": 18,
    "Inequality": 19,
    "Long Covid": 7,
    "Mechanism": 3,
    "Medical Devices": 20,
    "Misinformation": 21,
    "Model Systems & Tools": 22,
    "Molecular Biology": 23,
    "Non-human": 24,
    "Non-medical": 25,
    "Pediatrics": 26,
    "Prevalence": 27,
    "Prevention": 2,
    "Psychology": 28,
    "Recommendations": 29,
    "Risk Factors": 30,
    "Surveillance": 31,
    "Transmission": 5,
    "Treatment": 1,
    "Vaccines": 32
  },
  "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
}