| { |
| "cfg": { |
| "jsonl_path": "/kaggle/input/main-data02/gnn_dataset_fixed.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.2568123483703105, |
| "use_rgcn": true, |
| "rgcn_num_bases": 0, |
| "lambda_ans": 0.5, |
| "lambda_rank": 0.1, |
| "hard_neg_k": 8, |
| "calib_max_nodes": 200000, |
| "rank_margin": 0.2, |
| "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": [ |
| "__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" |
| ] |
| } |
