{ "edges": [ { "id": "Input: graph edges 1 Graph conv 1", "source": "Input: graph edges 1", "sourceHandle": "edges", "target": "Graph conv 1", "targetHandle": "edges" }, { "id": "Graph conv 1 Activation 1", "source": "Graph conv 1", "sourceHandle": "x", "target": "Activation 1", "targetHandle": "x" }, { "id": "Repeat 1 Graph conv 1", "source": "Repeat 1", "sourceHandle": "output", "target": "Graph conv 1", "targetHandle": "x" }, { "id": "Input: tensor 1 Graph conv 1", "source": "Input: tensor 1", "sourceHandle": "output", "target": "Graph conv 1", "targetHandle": "x" }, { "id": "MSE loss 1 Optimizer 1", "source": "MSE loss 1", "sourceHandle": "output", "target": "Optimizer 1", "targetHandle": "loss" }, { "id": "Activation 1 output Repeat 1 input", "source": "Activation 1", "sourceHandle": "output", "target": "Repeat 1", "targetHandle": "input" }, { "id": "LSTM 1 output Concatenate 1 b", "source": "LSTM 1", "sourceHandle": "output", "target": "Concatenate 1", "targetHandle": "b" }, { "id": "Concatenate 1 output Neural ODE with MLP 1 x", "source": "Concatenate 1", "sourceHandle": "output", "target": "Neural ODE with MLP 1", "targetHandle": "x" }, { "id": "Input: sequential 3 y LSTM 1 x", "source": "Input: sequential 3", "sourceHandle": "y", "target": "LSTM 1", "targetHandle": "x" }, { "id": "Input: sequential 1 y MSE loss 1 y", "source": "Input: sequential 1", "sourceHandle": "y", "target": "MSE loss 1", "targetHandle": "y" }, { "id": "Input: sequential 2 y Neural ODE with MLP 1 t", "source": "Input: sequential 2", "sourceHandle": "y", "target": "Neural ODE with MLP 1", "targetHandle": "t" }, { "id": "Input: tensor 2 output Neural ODE with MLP 1 y0", "source": "Input: tensor 2", "sourceHandle": "output", "target": "Neural ODE with MLP 1", "targetHandle": "y0" }, { "id": "Output 1 x MSE loss 1 x", "source": "Output 1", "sourceHandle": "x", "target": "MSE loss 1", "targetHandle": "x" }, { "id": "Neural ODE with MLP 1 y Output 1 x", "source": "Neural ODE with MLP 1", "sourceHandle": "y", "target": "Output 1", "targetHandle": "x" }, { "id": "Linear 1 output Concatenate 1 a", "source": "Linear 1", "sourceHandle": "output", "target": "Concatenate 1", "targetHandle": "a" }, { "id": "Concatenate 2 output Linear 1 x", "source": "Concatenate 2", "sourceHandle": "output", "target": "Linear 1", "targetHandle": "x" }, { "id": "Input: graph edges 2 edges Graph conv 2 edges", "source": "Input: graph edges 2", "sourceHandle": "edges", "target": "Graph conv 2", "targetHandle": "edges" }, { "id": "Input: tensor 3 output Embedding 1 x", "source": "Input: tensor 3", "sourceHandle": "output", "target": "Embedding 1", "targetHandle": "x" }, { "id": "Input: tensor 4 output Embedding 2 x", "source": "Input: tensor 4", "sourceHandle": "output", "target": "Embedding 2", "targetHandle": "x" }, { "id": "Activation 1 output Heterogeneous graph conv 1 node_embeddings", "source": "Activation 1", "sourceHandle": "output", "target": "Heterogeneous graph conv 1", "targetHandle": "node_embeddings" }, { "id": "Embedding 2 output Heterogeneous graph conv 1 node_embeddings", "source": "Embedding 2", "sourceHandle": "output", "target": "Heterogeneous graph conv 1", "targetHandle": "node_embeddings" }, { "id": "Embedding 1 output Heterogeneous graph conv 1 node_embeddings", "source": "Embedding 1", "sourceHandle": "output", "target": "Heterogeneous graph conv 1", "targetHandle": "node_embeddings" }, { "id": "Input: graph edges 3 edges Graph conv 3 edges", "source": "Input: graph edges 3", "sourceHandle": "edges", "target": "Graph conv 3", "targetHandle": "edges" }, { "id": "Graph conv 3 x Heterogeneous graph conv 1 edge_modules", "source": "Graph conv 3", "sourceHandle": "x", "target": "Heterogeneous graph conv 1", "targetHandle": "edge_modules" }, { "id": "Graph conv 2 x Heterogeneous graph conv 1 edge_modules", "source": "Graph conv 2", "sourceHandle": "x", "target": "Heterogeneous graph conv 1", "targetHandle": "edge_modules" }, { "id": "Heterogeneous graph conv 1 x Pick element by constant 1 x", "source": "Heterogeneous graph conv 1", "sourceHandle": "x", "target": "Pick element by constant 1", "targetHandle": "x" }, { "id": "Heterogeneous graph conv 1 x Pick element by constant 2 x", "source": "Heterogeneous graph conv 1", "sourceHandle": "x", "target": "Pick element by constant 2", "targetHandle": "x" }, { "id": "Heterogeneous graph conv 1 x Pick element by constant 3 x", "source": "Heterogeneous graph conv 1", "sourceHandle": "x", "target": "Pick element by constant 3", "targetHandle": "x" }, { "id": "Pick element by constant 2 x_i Mean pool 3 x", "source": "Pick element by constant 2", "sourceHandle": "x_i", "target": "Mean pool 3", "targetHandle": "x" }, { "id": "Mean pool 3 output Concatenate 2 b", "source": "Mean pool 3", "sourceHandle": "output", "target": "Concatenate 2", "targetHandle": "b" }, { "id": "Concatenate 3 output Concatenate 2 a", "source": "Concatenate 3", "sourceHandle": "output", "target": "Concatenate 2", "targetHandle": "a" }, { "id": "Pick element by constant 3 x_i Mean pool 1 x", "source": "Pick element by constant 3", "sourceHandle": "x_i", "target": "Mean pool 1", "targetHandle": "x" }, { "id": "Pick element by constant 1 x_i Mean pool 2 x", "source": "Pick element by constant 1", "sourceHandle": "x_i", "target": "Mean pool 2", "targetHandle": "x" }, { "id": "Mean pool 2 output Concatenate 3 b", "source": "Mean pool 2", "sourceHandle": "output", "target": "Concatenate 3", "targetHandle": "b" }, { "id": "Mean pool 1 output Concatenate 3 a", "source": "Mean pool 1", "sourceHandle": "output", "target": "Concatenate 3", "targetHandle": "a" } ], "env": "PyTorch model", "nodes": [ { "data": { "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": {}, "status": "done", "title": "Group" }, "height": 2361.0, "id": "Group 4", "position": { "x": -1553.0193773500205, "y": -240.79204628408627 }, "type": "node_group", "width": 2067.0 }, { "data": { "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": {}, "status": "done", "title": "Group" }, "height": 801.0, "id": "Group 5", "position": { "x": 653.8616229610058, "y": -2144.434037892992 }, "type": "node_group", "width": 686.0 }, { "data": { "__execution_delay": null, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "color": "purple" }, "status": "done", "title": "Group" }, "extent": "parent", "height": 616.0, "id": "Group 3", "parentId": "Group 4", "position": { "x": 955.0773075119091, "y": 276.48133806627845 }, "selected": false, "type": "node_group", "width": 1035.0 }, { "data": { "__execution_delay": null, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "color": "green" }, "status": "done", "title": "Group" }, "extent": "parent", "height": 1042.0, "id": "Group 2", "parentId": "Group 4", "position": { "x": 10.0, "y": 1309.1540753489603 }, "selected": false, "type": "node_group", "width": 751.0 }, { "data": { "__execution_delay": null, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "color": "blue" }, "status": "done", "title": "Group" }, "height": 793.0, "id": "Group 1", "position": { "x": 568.8655324518227, "y": -238.1553730513866 }, "type": "node_group", "width": 420.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } }, { "name": "edges", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Graph conv", "outputs": [ { "name": "x", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "GCNConv", "name": "type", "type": { "enum": [ "GCNConv", "GATConv", "GATv2Conv", "SAGEConv" ] } } ], "type": "basic" }, "params": { "type": "GCNConv" }, "status": "done", "title": "Graph conv" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Graph conv 1", "parentId": "Group 2", "position": { "x": 382.5829138643958, "y": 496.96416907011553 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "input", "position": "top", "type": { "type": "tensor" } } ], "name": "Repeat", "outputs": [ { "name": "output", "position": "bottom", "type": { "type": "tensor" } } ], "params": [ { "default": 1.0, "name": "times", "type": { "type": "" } }, { "default": false, "name": "same_weights", "type": { "type": "" } } ], "type": "basic" }, "params": { "times": "2" }, "status": "done", "title": "Repeat" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Repeat 1", "parentId": "Group 2", "position": { "x": 84.9020539861067, "y": 360.31211016780753 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "a", "position": "bottom", "type": { "type": "" } }, { "name": "b", "position": "bottom", "type": { "type": "" } } ], "name": "Concatenate", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Concatenate" }, "dragHandle": ".drag-handle", "height": 200.0, "id": "Concatenate 1", "position": { "x": 477.88148637482334, "y": -372.62774030487003 }, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": false, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: graph edges", "outputs": [ { "name": "edges", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "gene-interacts-gene" }, "status": "done", "title": "Input: graph edges" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 179.0, "id": "Input: graph edges 1", "parentId": "Group 2", "position": { "x": 486.4299294450075, "y": 836.0601064030513 }, "selected": false, "type": "basic", "width": 255.0 }, { "data": { "__execution_delay": null, "collapsed": false, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Activation", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [ { "default": "ReLU", "name": "type", "type": { "enum": [ "ReLU", "Leaky_ReLU", "Tanh", "Mish" ] } } ], "type": "basic" }, "params": { "type": 1.0 }, "status": "done", "title": "Activation" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Activation 1", "parentId": "Group 2", "position": { "x": 383.42692421440324, "y": 158.68687593857499 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": false, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "LSTM", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [ { "default": 1024.0, "name": "input_size", "type": { "type": "" } }, { "default": 1024.0, "name": "hidden_size", "type": { "type": "" } }, { "default": 0.0, "name": "dropout", "type": { "type": "" } } ], "type": "basic" }, "params": { "dropout": "0.2", "hidden_size": "128", "input_size": "128" }, "status": "done", "title": "LSTM" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 312.0, "id": "LSTM 1", "parentId": "Group 1", "position": { "x": 28.315247542851353, "y": 25.635423911301217 }, "selected": false, "type": "basic", "width": 204.0 }, { "data": { "__execution_delay": 0.0, "collapsed": false, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: sequential", "outputs": [ { "name": "y", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "future volumes " }, "status": "done", "title": "Input: sequential" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 176.0, "id": "Input: sequential 1", "parentId": "Group 5", "position": { "x": 314.61896315858814, "y": 614.6995504433646 }, "selected": false, "type": "basic", "width": 230.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } }, { "name": "y", "position": "bottom", "type": { "type": "" } } ], "name": "MSE loss", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "MSE loss" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "MSE loss 1", "parentId": "Group 5", "position": { "x": 10.0, "y": 369.6042243521049 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "green", "doc": null, "inputs": [ { "name": "loss", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Optimizer", "outputs": [], "params": [ { "default": "AdamW", "name": "type", "type": { "enum": [ "AdamW", "Adafactor", "Adagrad", "SGD", "Lion", "Paged AdamW", "Galore AdamW" ] } }, { "default": 0.0001, "name": "lr", "type": { "type": "" } } ], "type": "basic" }, "params": { "lr": "0.0001", "type": "AdamW" }, "status": "done", "title": "Optimizer" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 247.0, "id": "Optimizer 1", "parentId": "Group 5", "position": { "x": 10.380437966635554, "y": 10.0 }, "selected": false, "type": "basic", "width": 190.0 }, { "data": { "__execution_delay": 0.0, "collapsed": false, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: tensor", "outputs": [ { "name": "output", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "gene embedding" }, "status": "done", "title": "Input: tensor" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Input: tensor 1", "parentId": "Group 2", "position": { "x": 260.415499213693, "y": 831.8305281333035 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: sequential", "outputs": [ { "name": "y", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "future timestamps" }, "status": "done", "title": "Input: sequential" }, "dragHandle": ".drag-handle", "height": 206.0, "id": "Input: sequential 2", "position": { "x": 1248.253547891751, "y": -168.41648271530312 }, "type": "basic", "width": 237.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } }, { "name": "y0", "position": "bottom", "type": { "type": "tensor" } }, { "name": "t", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Neural ODE with MLP", "outputs": [ { "name": "y", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "dopri8", "name": "method", "type": { "enum": [ "dopri8", "dopri5", "bosh3", "fehlberg2", "adaptive_heun", "euler", "midpoint", "rk4", "explicit_adams", "implicit_adams" ] } }, { "default": null, "name": "relative_tolerance", "type": { "type": "None" } }, { "default": null, "name": "absolute_tolerance", "type": { "type": "None" } }, { "default": null, "name": "mlp_layers", "type": { "type": "None" } }, { "default": null, "name": "mlp_hidden_size", "type": { "type": "None" } }, { "default": "ReLU", "name": "mlp_activation", "type": { "enum": [ "ReLU", "Tanh", "Sigmoid" ] } } ], "type": "basic" }, "params": { "absolute_tolerance": "0.001", "method": "dopri8", "mlp_activation": "ReLU", "mlp_hidden_size": "64", "mlp_layers": "3", "relative_tolerance": " 0.001 " }, "status": "done", "title": "Neural ODE with MLP" }, "dragHandle": ".drag-handle", "height": 531.0, "id": "Neural ODE with MLP 1", "position": { "x": 592.4620140488937, "y": -991.7169229426221 }, "type": "basic", "width": 277.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: sequential", "outputs": [ { "name": "y", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "past volumes" }, "status": "done", "title": "Input: sequential" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Input: sequential 3", "parentId": "Group 1", "position": { "x": 32.39864750027925, "y": 583.4683505178409 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: tensor", "outputs": [ { "name": "output", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "current volume " }, "status": "done", "title": "Input: tensor" }, "dragHandle": ".drag-handle", "height": 200.0, "id": "Input: tensor 2", "position": { "x": 1033.7785231717685, "y": -160.89202606313793 }, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Output", "outputs": [ { "name": "x", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "future volume predictions" }, "status": "done", "title": "Output" }, "dragHandle": ".drag-handle", "height": 155.0, "id": "Output 1", "position": { "x": 575.2153360305078, "y": -1217.3477063035696 }, "type": "basic", "width": 304.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Linear", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [ { "default": 1024.0, "name": "output_dim", "type": { "type": "" } } ], "type": "basic" }, "params": { "output_dim": "256" }, "status": "done", "title": "Linear" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Linear 1", "parentId": "Group 4", "position": { "x": 1798.160952252842, "y": 10.0 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "a", "position": "bottom", "type": { "type": "" } }, { "name": "b", "position": "bottom", "type": { "type": "" } } ], "name": "Concatenate", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Concatenate" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Concatenate 2", "parentId": "Group 3", "position": { "x": 751.8645969652298, "y": 0.0 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "a", "position": "bottom", "type": { "type": "" } }, { "name": "b", "position": "bottom", "type": { "type": "" } } ], "name": "Concatenate", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Concatenate" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Concatenate 3", "parentId": "Group 3", "position": { "x": 467.59424395826454, "y": 175.01732824928223 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } }, { "name": "edges", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Graph conv", "outputs": [ { "name": "x", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "GCNConv", "name": "type", "type": { "enum": [ "GCNConv", "GATConv", "GATv2Conv", "SAGEConv" ] } } ], "type": "basic" }, "params": { "type": "GATConv" }, "status": "done", "title": "Graph conv" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Graph conv 2", "parentId": "Group 4", "position": { "x": 1806.181634199722, "y": 1581.8135304557259 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: tensor", "outputs": [ { "name": "output", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "drug index" }, "status": "done", "title": "Input: tensor" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Input: tensor 3", "parentId": "Group 4", "position": { "x": 1105.957400218611, "y": 2144.0418184175396 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: graph edges", "outputs": [ { "name": "edges", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "drug-targets-gene" }, "status": "done", "title": "Input: graph edges" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Input: graph edges 2", "parentId": "Group 4", "position": { "x": 1856.9227554163308, "y": 2140.745265404792 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "node_embeddings", "position": "bottom", "type": { "type": "tensor" } }, { "name": "edge_modules", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Heterogeneous graph conv", "outputs": [ { "name": "x", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "node_embeddings_order", "type": { "type": "None" } }, { "default": null, "name": "edge_modules_order", "type": { "type": "None" } } ], "type": "basic" }, "params": { "edge_modules": "", "edge_modules_order": "disease-assoc-gene, drug-targets-gene", "node_embedding_order": "gene, disease, drug", "node_embeddings": "gene, disease, drug", "node_embeddings_order": "gene, disease, drug " }, "status": "done", "title": "Heterogeneous graph conv" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 226.0, "id": "Heterogeneous graph conv 1", "parentId": "Group 4", "position": { "x": 1149.3945937733813, "y": 1087.9142231093092 }, "selected": false, "type": "basic", "width": 719.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } }, { "name": "edges", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Graph conv", "outputs": [ { "name": "x", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "GCNConv", "name": "type", "type": { "enum": [ "GCNConv", "GATConv", "GATv2Conv", "SAGEConv" ] } } ], "type": "basic" }, "params": { "type": "GATConv" }, "status": "done", "title": "Graph conv" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Graph conv 3", "parentId": "Group 4", "position": { "x": 1533.5142756639366, "y": 1575.5014666775842 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Embedding", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [ { "default": null, "name": "num_embeddings", "type": { "type": "" } }, { "default": null, "name": "embedding_dim", "type": { "type": "" } } ], "type": "basic" }, "params": { "embedding_dim": "64 ", "num_embeddings": "6" }, "status": "done", "title": "Embedding" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 227.0, "id": "Embedding 1", "parentId": "Group 4", "position": { "x": 1103.9876919196581, "y": 1567.257213663298 }, "selected": false, "type": "basic", "width": 202.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "blue", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Embedding", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [ { "default": null, "name": "num_embeddings", "type": { "type": "" } }, { "default": null, "name": "embedding_dim", "type": { "type": "" } } ], "type": "basic" }, "params": { "embedding_dim": "64", "num_embeddings": "6" }, "status": "done", "title": "Embedding" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 237.0, "id": "Embedding 2", "parentId": "Group 4", "position": { "x": 840.8658859321138, "y": 1555.8526944870114 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: tensor", "outputs": [ { "name": "output", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "disease index" }, "status": "done", "title": "Input: tensor" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Input: tensor 4", "parentId": "Group 4", "position": { "x": 839.4611346514839, "y": 2150.4203718536514 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "gray", "doc": null, "inputs": [], "name": "Input: graph edges", "outputs": [ { "name": "edges", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": null, "name": "name", "type": { "type": "None" } } ], "type": "basic" }, "params": { "name": "disease-assoc-gene" }, "status": "done", "title": "Input: graph edges" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 201.0, "id": "Input: graph edges 3", "parentId": "Group 4", "position": { "x": 1575.5522538981563, "y": 2134.427589093947 }, "selected": false, "type": "basic", "width": 225.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Pick element by constant", "outputs": [ { "name": "x_i", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "0", "name": "index", "type": { "type": "" } } ], "type": "basic" }, "params": { "index": "disease " }, "status": "done", "title": "Pick element by constant" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Pick element by constant 1", "parentId": "Group 3", "position": { "x": 478.1215695143337, "y": 403.19328354072366 }, "selected": false, "type": "basic", "width": 245.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Pick element by constant", "outputs": [ { "name": "x_i", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "0", "name": "index", "type": { "type": "" } } ], "type": "basic" }, "params": { "index": "drug" }, "status": "done", "title": "Pick element by constant" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Pick element by constant 2", "parentId": "Group 3", "position": { "x": 754.9659372119527, "y": 405.6468831887491 }, "selected": false, "type": "basic", "width": 270.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "tensor" } } ], "name": "Pick element by constant", "outputs": [ { "name": "x_i", "position": "top", "type": { "type": "tensor" } } ], "params": [ { "default": "0", "name": "index", "type": { "type": "" } } ], "type": "basic" }, "params": { "index": "gene" }, "status": "done", "title": "Pick element by constant" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 197.0, "id": "Pick element by constant 3", "parentId": "Group 3", "position": { "x": 197.120167516704, "y": 403.5288928546231 }, "selected": false, "type": "basic", "width": 247.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Mean pool", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Mean pool" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Mean pool 1", "parentId": "Group 3", "position": { "x": 221.58363816086091, "y": 304.61351982713296 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Mean pool", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Mean pool" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Mean pool 2", "parentId": "Group 3", "position": { "x": 500.14194620596317, "y": 301.77073327896215 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": null, "collapsed": true, "display": null, "error": null, "input_metadata": null, "meta": { "color": "orange", "doc": null, "inputs": [ { "name": "x", "position": "bottom", "type": { "type": "" } } ], "name": "Mean pool", "outputs": [ { "name": "output", "position": "top", "type": { "type": "None" } } ], "params": [], "type": "basic" }, "params": {}, "status": "done", "title": "Mean pool" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 200.0, "id": "Mean pool 3", "parentId": "Group 3", "position": { "x": 785.4145357661607, "y": 300.1249006428384 }, "selected": false, "type": "basic", "width": 200.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Past tumor volume embedding with LSTM" }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 130.0, "id": "Comment 1", "parentId": "Group 1", "position": { "x": 10.0, "y": 406.3216561052855 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Gene-gene graph embedding" }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 90.0, "id": "Comment 2", "parentId": "Group 2", "position": { "x": 10.0, "y": 214.69868480627042 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Learned initial drug & disease embeddings" }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 90.0, "id": "Comment 3", "parentId": "Group 4", "position": { "x": 886.9430712292642, "y": 1852.533426575776 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Extracting the graph embedding\n\nWe mean-pool the node embeddings of each node type, then concatenate these vectors." }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 178.0, "id": "Comment 4", "parentId": "Group 3", "position": { "x": 10.0, "y": 141.970585886959 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Knowledge graph embedding with GNN\n\nThis heterogeneous graph convolution creates node embeddings for all genes, diseases, and drugs in the graph." }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 206.0, "id": "Comment 5", "parentId": "Group 4", "position": { "x": 686.3753672347915, "y": 977.8926090141683 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Tumor volume trajectory predictions with ODE\n\nAn MLP is used to predict the derivative of the tumor volume at different stages, based on the knowledge graph embedding and the past tumor volume embeddings." }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "height": 274.0, "id": "Comment 6", "position": { "x": -0.7721924881518737, "y": -865.6155529412208 }, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "# Training configuration\n\nThe loss for the whole model is simply the MSE loss over the future volume measurements." }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 138.0, "id": "Comment 7", "parentId": "Group 5", "position": { "x": 276.18914557042024, "y": 267.8721350317928 }, "selected": false, "type": "comment", "width": 400.0 }, { "data": { "__execution_delay": 0.0, "collapsed": null, "display": null, "error": "Unknown operation.", "input_metadata": null, "meta": {}, "params": { "text": "## Disease-gene and drug-gene edges" }, "status": "done", "title": "Comment" }, "dragHandle": ".drag-handle", "extent": "parent", "height": 74.0, "id": "Comment 8", "parentId": "Group 4", "position": { "x": 1615.9956563920114, "y": 1872.1215872934476 }, "selected": false, "type": "comment", "width": 400.0 } ] }