{
  "cfg": {
    "jsonl_path": "/kaggle/input/final-data/gnn_dataset_merged.json",
    "node_emb_npz": "/kaggle/input/node-embeddings01/node_embeds_ada002.npz",
    "q_encoder_name": "distilbert-base-uncased",
    "max_q_len": 128,
    "freeze_q_encoder": true,
    "hidden_dim": 256,
    "instr_k": 3,
    "gnn_layers": 3,
    "revise_T": 2,
    "max_hop": 4,
    "batch_size": 8,
    "lr": 0.0002,
    "weight_decay": 0.0001,
    "epochs": 8,
    "grad_clip": 1.0,
    "use_lr_scheduler": true,
    "lr_plateau_factor": 0.5,
    "lr_plateau_patience": 1,
    "min_lr": 1e-06,
    "early_stop_patience": 2,
    "early_stop_min_delta": 0.0,
    "use_focal": false,
    "focal_gamma": 2.0,
    "pos_weight": 0.434991974317817,
    "use_rgcn": true,
    "rgcn_num_bases": 0,
    "w_ans": 0.5,
    "w_rank": 0.1,
    "hard_neg_k": 8,
    "calib_max_nodes": 200000,
    "rank_margin": 0.2,
    "phase1_epochs": 3,
    "use_balanced_phase2": true,
    "w_node": 1.0,
    "use_listwise": false,
    "w_listwise": 0.25,
    "use_ans_pos_weight": true,
    "out_dir": "/kaggle/working/rearev_nodeclf",
    "save_name": "model.pt",
    "test_jsonl_path": "/kaggle/input/testin-data-reduced/evaluation_dataset_processed.jsonl",
    "infer_topk": 10,
    "tau_ans": 0.5,
    "infer_tau_ans": 0.5,
    "infer_tau_node": 0.5,
    "temp_ans": 1.0,
    "temp_node": 1.0,
    "pred_path": "/kaggle/working/predictions.jsonl",
    "drop_unlabeled": true
  },
  "rel_list": [
    "__UNK__",
    "Compound-binds-Gene",
    "Compound-downregulates-Gene",
    "Compound-resembles-Compound",
    "Compound-upregulates-Gene",
    "Gene-covaries-Gene",
    "Gene-interacts-Gene",
    "Gene-participates-Biological Process",
    "Gene-participates-Molecular Function",
    "Gene-participates-Pathway",
    "Gene-regulates-Gene",
    "__REV__Compound-binds-Gene",
    "__REV__Compound-downregulates-Gene",
    "__REV__Compound-resembles-Compound",
    "__REV__Compound-upregulates-Gene",
    "__REV__Gene-covaries-Gene",
    "__REV__Gene-interacts-Gene",
    "__REV__Gene-participates-Biological Process",
    "__REV__Gene-participates-Molecular Function",
    "__REV__Gene-participates-Pathway",
    "__REV__Gene-regulates-Gene"
  ]
}