config.json

{
  "classes": [
    "I",
    "II",
    "III",
    "IV",
    "V",
    "VI",
    "VII",
    "VIII",
    "IX",
    "X",
    "XI",
    "XII",
    "XIII",
    "XIV",
    "XV",
    "XVI",
    "XVII",
    "XVIII",
    "XIX",
    "XX",
    "XXI",
    "XXII"
  ],
  "dropout": 0.15,
  "hidden_sizes": [
    256
  ],
  "input_dim": 768,
  "model_name": "models/pubmedbert-base-embeddings"
}