lynxkite / examples /ODE-GNN.lynxkite.json
darabos's picture
More detailed ODE-GNN mock-up.
6cff9d9
raw
history blame
51.3 kB
{
"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": "Graph conv 2 x Heterogeneous graph conv 1 cases",
"source": "Graph conv 2",
"sourceHandle": "x",
"target": "Heterogeneous graph conv 1",
"targetHandle": "cases"
},
{
"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": {
"__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": ".bg-primary",
"height": 200.0,
"id": "Graph conv 1",
"position": {
"x": -958.3664302023487,
"y": 1599.1160982851618
},
"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": "<class 'int'>"
}
},
{
"default": false,
"name": "same_weights",
"type": {
"type": "<class 'bool'>"
}
}
],
"type": "basic"
},
"params": {
"times": "2"
},
"status": "done",
"title": "Repeat"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Repeat 1",
"position": {
"x": -1256.0472900806378,
"y": 1462.4640393828538
},
"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": "<class 'inspect._empty'>"
}
},
{
"name": "b",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Concatenate",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Concatenate"
},
"dragHandle": ".bg-primary",
"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": ".bg-primary",
"height": 179.0,
"id": "Input: graph edges 1",
"position": {
"x": -854.519414621737,
"y": 1938.2120356180976
},
"type": "basic",
"width": 255.0
},
{
"data": {
"display": null,
"error": null,
"input_metadata": null,
"meta": {
"color": "orange",
"doc": null,
"inputs": [
{
"name": "x",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"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": ".bg-primary",
"height": 200.0,
"id": "Activation 1",
"position": {
"x": -957.5224198523413,
"y": 1334.1909972833182
},
"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": "<class 'inspect._empty'>"
}
}
],
"name": "LSTM",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [
{
"default": 1024.0,
"name": "input_size",
"type": {
"type": "<class 'int'>"
}
},
{
"default": 1024.0,
"name": "hidden_size",
"type": {
"type": "<class 'int'>"
}
},
{
"default": 0.0,
"name": "dropout",
"type": {
"type": "<class 'float'>"
}
}
],
"type": "basic"
},
"params": {
"dropout": "0.2",
"hidden_size": "128",
"input_size": "128"
},
"status": "done",
"title": "LSTM"
},
"dragHandle": ".bg-primary",
"height": 312.0,
"id": "LSTM 1",
"position": {
"x": 662.5652799873882,
"y": -47.63729698454539
},
"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": ".bg-primary",
"height": 176.0,
"id": "Input: sequential 1",
"position": {
"x": 1724.6836018444237,
"y": 397.2031221699466
},
"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": "<class 'inspect._empty'>"
}
},
{
"name": "y",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "MSE loss",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "MSE loss"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "MSE loss 1",
"position": {
"x": 784.6649975930936,
"y": -1425.9136836194843
},
"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": "<class 'float'>"
}
}
],
"type": "basic"
},
"params": {
"lr": "0.0001",
"type": "AdamW"
},
"status": "done",
"title": "Optimizer"
},
"dragHandle": ".bg-primary",
"height": 247.0,
"id": "Optimizer 1",
"position": {
"x": 792.7562892596496,
"y": -1782.9476234049491
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: tensor 1",
"position": {
"x": -1080.5338448530515,
"y": 1933.9824573483497
},
"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": ".bg-primary",
"height": 206.0,
"id": "Input: sequential 2",
"position": {
"x": 1404.6077870047632,
"y": 393.03737591778565
},
"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": ".bg-primary",
"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": ".bg-primary",
"height": 200.0,
"id": "Input: sequential 3",
"position": {
"x": 848.5881581854119,
"y": 393.6406513741127
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: tensor 2",
"position": {
"x": 1103.4272296857464,
"y": 383.50500648489503
},
"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": ".bg-primary",
"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": "<class 'inspect._empty'>"
}
}
],
"name": "Linear",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [
{
"default": 1024.0,
"name": "output_dim",
"type": {
"type": "<class 'int'>"
}
}
],
"type": "basic"
},
"params": {
"output_dim": "256"
},
"status": "done",
"title": "Linear"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Linear 1",
"position": {
"x": 260.1415749028216,
"y": -225.79204628408624
},
"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": "<class 'inspect._empty'>"
}
},
{
"name": "b",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Concatenate",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Concatenate"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Concatenate 2",
"position": {
"x": 199.77501256646406,
"y": 50.6892917821922
},
"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": "<class 'inspect._empty'>"
}
},
{
"name": "b",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Concatenate",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Concatenate"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Concatenate 3",
"position": {
"x": -207.83562912506517,
"y": 177.1410132322926
},
"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": ".bg-primary",
"height": 200.0,
"id": "Graph conv 2",
"position": {
"x": 568.0618872822669,
"y": 1248.5801580938696
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: tensor 3",
"position": {
"x": -237.06197713140955,
"y": 1913.2497721334535
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: graph edges 2",
"position": {
"x": 598.9033780663102,
"y": 1889.9532191207059
},
"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": ".bg-primary",
"height": 226.0,
"id": "Heterogeneous graph conv 1",
"position": {
"x": -193.62478357663917,
"y": 857.1221768252228
},
"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": ".bg-primary",
"height": 200.0,
"id": "Graph conv 3",
"position": {
"x": 196.18050700893468,
"y": 1250.8968769256908
},
"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": "<class 'inspect._empty'>"
}
}
],
"name": "Embedding",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [
{
"default": null,
"name": "num_embeddings",
"type": {
"type": "<class 'int'>"
}
},
{
"default": null,
"name": "embedding_dim",
"type": {
"type": "<class 'int'>"
}
}
],
"type": "basic"
},
"params": {
"embedding_dim": "64",
"num_embeddings": "6"
},
"status": "done",
"title": "Embedding"
},
"dragHandle": ".bg-primary",
"height": 227.0,
"id": "Embedding 1",
"position": {
"x": -256.0885115154183,
"y": 1322.2511456416655
},
"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": "<class 'inspect._empty'>"
}
}
],
"name": "Embedding",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [
{
"default": null,
"name": "num_embeddings",
"type": {
"type": "<class 'int'>"
}
},
{
"default": null,
"name": "embedding_dim",
"type": {
"type": "<class 'int'>"
}
}
],
"type": "basic"
},
"params": {
"embedding_dim": "64",
"num_embeddings": "6"
},
"status": "done",
"title": "Embedding"
},
"dragHandle": ".bg-primary",
"height": 237.0,
"id": "Embedding 2",
"position": {
"x": -502.1534914179067,
"y": 1325.0606482029252
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: tensor 4",
"position": {
"x": -503.55824269853656,
"y": 1919.6283255695653
},
"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": ".bg-primary",
"height": 200.0,
"id": "Input: graph edges 3",
"position": {
"x": 232.53287654813585,
"y": 1903.635542809861
},
"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": "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": "<class 'str'>"
}
}
],
"type": "basic"
},
"params": {
"index": "disease"
},
"status": "done",
"title": "Pick element by constant"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Pick element by constant 1",
"position": {
"x": -104.82050032377765,
"y": 443.88257532291584
},
"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": "<class 'str'>"
}
}
],
"type": "basic"
},
"params": {
"index": "drug"
},
"status": "done",
"title": "Pick element by constant"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Pick element by constant 2",
"position": {
"x": 172.0238673738413,
"y": 446.3361749709413
},
"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": "<class 'str'>"
}
}
],
"type": "basic"
},
"params": {
"index": "gene"
},
"status": "done",
"title": "Pick element by constant"
},
"dragHandle": ".bg-primary",
"height": 197.0,
"id": "Pick element by constant 3",
"position": {
"x": -385.8219023214074,
"y": 444.2181846368153
},
"type": "basic",
"width": 247.0
},
{
"data": {
"__execution_delay": null,
"collapsed": true,
"display": null,
"error": null,
"input_metadata": null,
"meta": {
"color": "blue",
"doc": null,
"inputs": [
{
"name": "x",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Mean pool",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Mean pool"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Mean pool 1",
"position": {
"x": -363.34092335643146,
"y": 338.36409073219164
},
"type": "basic",
"width": 200.0
},
{
"data": {
"__execution_delay": null,
"collapsed": true,
"display": null,
"error": null,
"input_metadata": null,
"meta": {
"color": "blue",
"doc": null,
"inputs": [
{
"name": "x",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Mean pool",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Mean pool"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Mean pool 2",
"position": {
"x": -82.8001236321482,
"y": 319.6613707505728
},
"type": "basic",
"width": 200.0
},
{
"data": {
"__execution_delay": null,
"collapsed": true,
"display": null,
"error": null,
"input_metadata": null,
"meta": {
"color": "blue",
"doc": null,
"inputs": [
{
"name": "x",
"position": "bottom",
"type": {
"type": "<class 'inspect._empty'>"
}
}
],
"name": "Mean pool",
"outputs": [
{
"name": "output",
"position": "top",
"type": {
"type": "None"
}
}
],
"params": [],
"type": "basic"
},
"params": {},
"status": "done",
"title": "Mean pool"
},
"dragHandle": ".bg-primary",
"height": 200.0,
"id": "Mean pool 3",
"position": {
"x": 201.4812200884588,
"y": 314.0505547560871
},
"type": "basic",
"width": 200.0
}
]
}