| { |
| "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_focal": true, |
| "focal_gamma": 2.0, |
| "pos_weight": 6.863562091503268, |
| "use_rgcn": true, |
| "rgcn_num_bases": 0, |
| "lambda_ans": 0.0, |
| "lambda_rank": 0.0, |
| "hard_neg_k": 8, |
| "calib_max_nodes": 200000, |
| "rank_margin": 0.2, |
| "use_answerability_head": false, |
| "use_listwise": true, |
| "listwise_num_neg": 128, |
| "hard_bank_k": 32, |
| "use_hard_bank": true, |
| "lambda_node": 0.25, |
| "lambda_listwise": 1.0, |
| "lambda_set": 0.5, |
| "lambda_cal": 0.05, |
| "lambda_div": 0.01, |
| "balance_multiseed": true, |
| "weight_multiseed": 2.0, |
| "weight_multigold": 1.5, |
| "use_film": true, |
| "use_step_pe": true, |
| "dropedge_p": 0.1, |
| "dyn_alpha": 0.6, |
| "dyn_min_k": 1, |
| "dyn_max_k": 20, |
| "tau_none": 0.35, |
| "out_dir": "/kaggle/working/rearev_nodeclf", |
| "save_name": "model.pt", |
| "test_jsonl_path": "/kaggle/input/testing-data1/evaluation_dataset_processed.jsonl", |
| "infer_topk": 10, |
| "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": [ |
| "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" |
| ], |
| "inference": { |
| "dyn_alpha": 0.6, |
| "dyn_min_k": 1, |
| "dyn_max_k": 20, |
| "tau_none": 0.35, |
| "temp_node": 1.0 |
| }, |
| "weights_file": "best_enhanced.pt" |
| } |
