| { | |
| "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" | |
| ] | |
| } |