{
  "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
    }
  ]
}