| { | |
| "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_focal": false, | |
| "focal_gamma": 2.0, | |
| "pos_weight": 4.147643384440658, | |
| "out_dir": "/kaggle/working/rearev_nodeclf", | |
| "save_name": "model.pt" | |
| }, | |
| "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" | |
| ] | |
| } |