config.json

{
  "model_type": "virtual_cell_patient",
  "architectures": ["VirtualCellPatientModel"],
  "auto_map": {
    "AutoConfig": "modeling_virtual_cell.VirtualCellPatientConfig",
    "AutoModel": "modeling_virtual_cell.VirtualCellPatientModel"
  },
  "n_genes": 18301,
  "embed_dim": 512,
  "hidden_dim": [4096, 1024],
  "dropout": 0.1,
  "residual": false,
  "activation": "prelu",
  "attention_hidden_dim": 512,
  "num_classes": 10,
  "classifier_dropout": 0.1,
  "id2label": {
    "0": "oncological",
    "1": "immune_inflammatory",
    "2": "neurological",
    "3": "metabolic_vascular",
    "4": "gastrointestinal",
    "5": "respiratory",
    "6": "epithelial_barrier",
    "7": "sensory_specialized",
    "8": "healthy_control",
    "9": "other"
  },
  "label2id": {
    "oncological": 0,
    "immune_inflammatory": 1,
    "neurological": 2,
    "metabolic_vascular": 3,
    "gastrointestinal": 4,
    "respiratory": 5,
    "epithelial_barrier": 6,
    "sensory_specialized": 7,
    "healthy_control": 8,
    "other": 9
  }
}