Spaces:
Running
Running
| { | |
| "edges": [ | |
| { | |
| "id": "Input: embedding 1 Graph conv 1", | |
| "source": "Input: embedding 1", | |
| "sourceHandle": "x", | |
| "target": "Graph conv 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "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": "Activation 1 Repeat 1", | |
| "source": "Activation 1", | |
| "sourceHandle": "x", | |
| "target": "Repeat 1", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Repeat 1 Graph conv 1", | |
| "source": "Repeat 1", | |
| "sourceHandle": "output", | |
| "target": "Graph conv 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Input: sequential 1 LSTM 1", | |
| "source": "Input: sequential 1", | |
| "sourceHandle": "y", | |
| "target": "LSTM 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Input: zeros 1 LSTM 1", | |
| "source": "Input: zeros 1", | |
| "sourceHandle": "x", | |
| "target": "LSTM 1", | |
| "targetHandle": "h" | |
| }, | |
| { | |
| "id": "Recurrent chain 1 LSTM 1", | |
| "source": "Recurrent chain 1", | |
| "sourceHandle": "output", | |
| "target": "LSTM 1", | |
| "targetHandle": "h" | |
| }, | |
| { | |
| "id": "LSTM 1 Recurrent chain 1", | |
| "source": "LSTM 1", | |
| "sourceHandle": "h", | |
| "target": "Recurrent chain 1", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Activation 1 Concatenate 1", | |
| "source": "Activation 1", | |
| "sourceHandle": "x", | |
| "target": "Concatenate 1", | |
| "targetHandle": "a" | |
| }, | |
| { | |
| "id": "LSTM 1 Concatenate 1", | |
| "source": "LSTM 1", | |
| "sourceHandle": "x", | |
| "target": "Concatenate 1", | |
| "targetHandle": "b" | |
| }, | |
| { | |
| "id": "Input: label 1 MSE loss 1", | |
| "source": "Input: label 1", | |
| "sourceHandle": "y", | |
| "target": "MSE loss 1", | |
| "targetHandle": "y" | |
| }, | |
| { | |
| "id": "MSE loss 1 Optimizer 1", | |
| "source": "MSE loss 1", | |
| "sourceHandle": "loss", | |
| "target": "Optimizer 1", | |
| "targetHandle": "loss" | |
| }, | |
| { | |
| "id": "Concatenate 1 Neural ODE 2", | |
| "source": "Concatenate 1", | |
| "sourceHandle": "x", | |
| "target": "Neural ODE 2", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Neural ODE 2 MSE loss 1", | |
| "source": "Neural ODE 2", | |
| "sourceHandle": "x", | |
| "target": "MSE loss 1", | |
| "targetHandle": "x" | |
| } | |
| ], | |
| "env": "PyTorch model", | |
| "nodes": [ | |
| { | |
| "data": { | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "edges": { | |
| "name": "edges", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| }, | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Graph conv", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "type": { | |
| "default": "1", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "GCNConv", | |
| "GATConv", | |
| "GATv2Conv", | |
| "SAGEConv" | |
| ] | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "type": 1.0 | |
| }, | |
| "status": "planned", | |
| "title": "Graph conv" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Graph conv 1", | |
| "position": { | |
| "x": 350.98078368755864, | |
| "y": 195.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "input": { | |
| "name": "input", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Repeat", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "times": { | |
| "default": 1.0, | |
| "name": "times", | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "times": "5" | |
| }, | |
| "status": "planned", | |
| "title": "Repeat" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Repeat 1", | |
| "position": { | |
| "x": -94.15168677219138, | |
| "y": 14.525356969883305 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "a": { | |
| "name": "a", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| }, | |
| "b": { | |
| "name": "b", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Concatenate", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "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": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: graph edges", | |
| "outputs": { | |
| "edges": { | |
| "name": "edges", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Input: graph edges" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: graph edges 1", | |
| "position": { | |
| "x": 515.6535517374441, | |
| "y": 545.4709559884296 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: embedding", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Input: embedding" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: embedding 1", | |
| "position": { | |
| "x": 246.6527948448857, | |
| "y": 551.6313504198322 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Activation", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "type": { | |
| "default": "1", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "ReLU", | |
| "LeakyReLU", | |
| "Tanh", | |
| "Mish" | |
| ] | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "type": 1.0 | |
| }, | |
| "status": "planned", | |
| "title": "Activation" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Activation 1", | |
| "position": { | |
| "x": 354.3731834561054, | |
| "y": -73.74768512965228 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "h": { | |
| "name": "h", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| }, | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "LSTM", | |
| "outputs": { | |
| "h": { | |
| "name": "h", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| }, | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "LSTM" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "LSTM 1", | |
| "position": { | |
| "x": 960.0, | |
| "y": 135.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: sequential", | |
| "outputs": { | |
| "y": { | |
| "name": "y", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Input: sequential" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: sequential 1", | |
| "position": { | |
| "x": 1005.0, | |
| "y": 510.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: zeros", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Input: zeros" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: zeros 1", | |
| "position": { | |
| "x": 1290.0, | |
| "y": 405.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "input": { | |
| "name": "input", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Recurrent chain", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Recurrent chain" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Recurrent chain 1", | |
| "position": { | |
| "x": 1224.6603040746108, | |
| "y": 135.44839862151363 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| }, | |
| "y": { | |
| "name": "y", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "MSE loss", | |
| "outputs": { | |
| "loss": { | |
| "name": "loss", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "MSE loss" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "MSE loss 1", | |
| "position": { | |
| "x": 915.0, | |
| "y": -900.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: label", | |
| "outputs": { | |
| "y": { | |
| "name": "y", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Input: label" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: label 1", | |
| "position": { | |
| "x": 1095.0, | |
| "y": -450.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "loss": { | |
| "name": "loss", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Optimizer", | |
| "outputs": {}, | |
| "params": { | |
| "lr": { | |
| "default": 0.001, | |
| "name": "lr", | |
| "type": { | |
| "type": "<class 'float'>" | |
| } | |
| }, | |
| "type": { | |
| "default": "1", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "AdamW", | |
| "Adafactor", | |
| "Adagrad", | |
| "SGD", | |
| "Lion", | |
| "Paged AdamW", | |
| "Galore AdamW" | |
| ] | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "lr": 0.001, | |
| "type": 1.0 | |
| }, | |
| "status": "planned", | |
| "title": "Optimizer" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 247.0, | |
| "id": "Optimizer 1", | |
| "position": { | |
| "x": 915.3430278730226, | |
| "y": -1268.0577550022126 | |
| }, | |
| "type": "basic", | |
| "width": 190.0 | |
| }, | |
| { | |
| "data": { | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Neural ODE", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "absolute_tolerance": { | |
| "default": null, | |
| "name": "absolute_tolerance", | |
| "type": { | |
| "type": "None" | |
| } | |
| }, | |
| "method": { | |
| "default": "1", | |
| "name": "method", | |
| "type": { | |
| "enum": [ | |
| "dopri8", | |
| "dopri5", | |
| "bosh3", | |
| "fehlberg2", | |
| "adaptive_heun", | |
| "euler", | |
| "midpoint", | |
| "rk4", | |
| "explicit_adams", | |
| "implicit_adams" | |
| ] | |
| } | |
| }, | |
| "relative_tolerance": { | |
| "default": null, | |
| "name": "relative_tolerance", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "absolute_tolerance": null, | |
| "method": 1.0, | |
| "relative_tolerance": null | |
| }, | |
| "status": "planned", | |
| "title": "Neural ODE" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Neural ODE 2", | |
| "position": { | |
| "x": 342.3226409443945, | |
| "y": -687.1882072175634 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| } | |
| ] | |
| } | |