Spaces:
Running
Running
| { | |
| "env": "PyTorch model", | |
| "nodes": [ | |
| { | |
| "id": "Input: features 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Input: features", | |
| "params": {}, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "name": "Input: features", | |
| "type": "basic", | |
| "outputs": { | |
| "x": { | |
| "position": "top", | |
| "name": "x", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "inputs": {} | |
| }, | |
| "collapsed": true, | |
| "__execution_delay": null | |
| }, | |
| "position": { | |
| "x": -108.60604658638658, | |
| "y": 63.96065124378427 | |
| }, | |
| "width": 200.0, | |
| "height": 200.0, | |
| "parentId": null | |
| }, | |
| { | |
| "id": "Input: graph edges 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Input: graph edges", | |
| "params": {}, | |
| "display": null, | |
| "error": null, | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "meta": { | |
| "name": "Input: graph edges", | |
| "inputs": {}, | |
| "params": {}, | |
| "type": "basic", | |
| "outputs": { | |
| "edges": { | |
| "name": "edges", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "top" | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 180.7373888617958, | |
| "y": 58.54904654355781 | |
| }, | |
| "width": 200.0, | |
| "parentId": null, | |
| "height": 200.0 | |
| }, | |
| { | |
| "id": "Linear 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Linear", | |
| "params": { | |
| "output_dim": "same" | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom", | |
| "name": "x" | |
| } | |
| }, | |
| "type": "basic", | |
| "name": "Linear", | |
| "outputs": { | |
| "x": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "top", | |
| "name": "x" | |
| } | |
| }, | |
| "params": { | |
| "output_dim": { | |
| "name": "output_dim", | |
| "type": { | |
| "type": "<class 'str'>" | |
| }, | |
| "default": "same" | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 78.37881963123723, | |
| "y": -528.3012263817914 | |
| }, | |
| "parentId": null, | |
| "width": 204.0, | |
| "height": 140.0 | |
| }, | |
| { | |
| "id": "Activation 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Activation", | |
| "params": { | |
| "type": "ReLU" | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "outputs": { | |
| "x": { | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "x" | |
| } | |
| }, | |
| "name": "Activation", | |
| "type": "basic", | |
| "inputs": { | |
| "x": { | |
| "position": "bottom", | |
| "name": "x", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "type": { | |
| "default": "OptionsFor_type.ReLU", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "ReLU", | |
| "LeakyReLU", | |
| "Tanh", | |
| "Mish" | |
| ] | |
| } | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 98.44658319023353, | |
| "y": -741.1411130550297 | |
| }, | |
| "height": 154.0, | |
| "parentId": null, | |
| "width": 173.0 | |
| }, | |
| { | |
| "id": "Dropout 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Dropout", | |
| "params": { | |
| "p": 0.5 | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "type": "basic", | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom" | |
| } | |
| }, | |
| "name": "Dropout", | |
| "params": { | |
| "p": { | |
| "default": 0.5, | |
| "type": { | |
| "type": "<class 'float'>" | |
| }, | |
| "name": "p" | |
| } | |
| }, | |
| "outputs": { | |
| "x": { | |
| "position": "top", | |
| "name": "x", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 73.61437458187964, | |
| "y": -929.9824215609918 | |
| }, | |
| "width": 207.0, | |
| "parentId": null, | |
| "height": 159.0 | |
| }, | |
| { | |
| "id": "Graph conv 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Graph conv", | |
| "params": { | |
| "type": "SAGEConv" | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "outputs": { | |
| "x": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "x", | |
| "position": "top" | |
| } | |
| }, | |
| "type": "basic", | |
| "name": "Graph conv", | |
| "params": { | |
| "type": { | |
| "name": "type", | |
| "default": "OptionsFor_type.GCNConv", | |
| "type": { | |
| "enum": [ | |
| "GCNConv", | |
| "GATConv", | |
| "GATv2Conv", | |
| "SAGEConv" | |
| ] | |
| } | |
| } | |
| }, | |
| "inputs": { | |
| "x": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "x", | |
| "position": "bottom" | |
| }, | |
| "edges": { | |
| "name": "edges", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom" | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 64.08886242755246, | |
| "y": -269.43023573181557 | |
| }, | |
| "height": 200.0, | |
| "width": 200.0, | |
| "parentId": null | |
| }, | |
| { | |
| "id": "Supervised loss 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Supervised loss", | |
| "params": {}, | |
| "display": null, | |
| "error": null, | |
| "collapsed": true, | |
| "__execution_delay": null, | |
| "meta": { | |
| "type": "basic", | |
| "params": {}, | |
| "name": "Supervised loss", | |
| "inputs": { | |
| "y": { | |
| "name": "y", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom" | |
| }, | |
| "x": { | |
| "name": "x", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom" | |
| } | |
| }, | |
| "outputs": { | |
| "loss": { | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "loss" | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 110.53693593362718, | |
| "y": -1123.9976567905628 | |
| }, | |
| "parentId": null, | |
| "height": 80.0, | |
| "width": 204.0 | |
| }, | |
| { | |
| "id": "Input: label 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Input: label", | |
| "params": {}, | |
| "display": null, | |
| "error": null, | |
| "collapsed": true, | |
| "__execution_delay": null, | |
| "meta": { | |
| "inputs": {}, | |
| "outputs": { | |
| "y": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "top", | |
| "name": "y" | |
| } | |
| }, | |
| "name": "Input: label", | |
| "type": "basic", | |
| "params": {} | |
| } | |
| }, | |
| "position": { | |
| "x": 666.110498676668, | |
| "y": -898.6721561114967 | |
| }, | |
| "width": 200.0, | |
| "height": 73.0, | |
| "parentId": null | |
| }, | |
| { | |
| "id": "Optimizer 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Optimizer", | |
| "params": { | |
| "lr": 0.001, | |
| "type": "AdamW" | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "name": "Optimizer", | |
| "outputs": {}, | |
| "params": { | |
| "lr": { | |
| "default": 0.001, | |
| "name": "lr", | |
| "type": { | |
| "type": "<class 'float'>" | |
| } | |
| }, | |
| "type": { | |
| "type": { | |
| "enum": [ | |
| "AdamW", | |
| "Adafactor", | |
| "Adagrad", | |
| "SGD", | |
| "Lion", | |
| "Paged AdamW", | |
| "Galore AdamW" | |
| ] | |
| }, | |
| "name": "type", | |
| "default": "OptionsFor_type.AdamW" | |
| } | |
| }, | |
| "inputs": { | |
| "loss": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "loss", | |
| "position": "bottom" | |
| } | |
| }, | |
| "type": "basic" | |
| } | |
| }, | |
| "position": { | |
| "x": 115.25730958703494, | |
| "y": -1431.5320892753364 | |
| }, | |
| "width": 200.0, | |
| "height": 233.0, | |
| "parentId": null | |
| }, | |
| { | |
| "id": "Repeat 3", | |
| "type": "basic", | |
| "data": { | |
| "title": "Repeat", | |
| "params": { | |
| "times": 1.0 | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "type": "basic", | |
| "position": { | |
| "y": 340.0, | |
| "x": 371.0 | |
| }, | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "bottom" | |
| } | |
| }, | |
| "name": "Repeat", | |
| "params": { | |
| "times": { | |
| "name": "times", | |
| "default": 1.0, | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| } | |
| }, | |
| "inputs": { | |
| "input": { | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "position": "top", | |
| "name": "input" | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": -245.1288628776232, | |
| "y": -276.90661040974317 | |
| }, | |
| "width": 200.0, | |
| "height": 200.0 | |
| }, | |
| { | |
| "id": "Repeat 1", | |
| "type": "basic", | |
| "data": { | |
| "title": "Repeat", | |
| "params": { | |
| "times": 1.0 | |
| }, | |
| "display": null, | |
| "error": null, | |
| "meta": { | |
| "outputs": { | |
| "output": { | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| }, | |
| "name": "output" | |
| } | |
| }, | |
| "params": { | |
| "times": { | |
| "default": 1.0, | |
| "type": { | |
| "type": "<class 'int'>" | |
| }, | |
| "name": "times" | |
| } | |
| }, | |
| "position": { | |
| "x": 387.0, | |
| "y": 337.0 | |
| }, | |
| "type": "basic", | |
| "name": "Repeat", | |
| "inputs": { | |
| "input": { | |
| "name": "input", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": -258.0088683218416, | |
| "y": -737.3822225246788 | |
| }, | |
| "width": 200.0, | |
| "height": 200.0 | |
| } | |
| ], | |
| "edges": [ | |
| { | |
| "id": "xy-edge__Linear 1x-Activation 1x", | |
| "source": "Linear 1", | |
| "target": "Activation 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "xy-edge__Activation 1x-Dropout 1x", | |
| "source": "Activation 1", | |
| "target": "Dropout 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "xy-edge__Input: features 1x-Graph conv 1x", | |
| "source": "Input: features 1", | |
| "target": "Graph conv 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "xy-edge__Input: graph edges 1edges-Graph conv 1edges", | |
| "source": "Input: graph edges 1", | |
| "target": "Graph conv 1", | |
| "sourceHandle": "edges", | |
| "targetHandle": "edges" | |
| }, | |
| { | |
| "id": "xy-edge__Graph conv 1x-Linear 1x", | |
| "source": "Graph conv 1", | |
| "target": "Linear 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "xy-edge__Input: label 1y-Supervised loss 1y", | |
| "source": "Input: label 1", | |
| "target": "Supervised loss 1", | |
| "sourceHandle": "y", | |
| "targetHandle": "y" | |
| }, | |
| { | |
| "id": "xy-edge__Dropout 1x-Supervised loss 1x", | |
| "source": "Dropout 1", | |
| "target": "Supervised loss 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "xy-edge__Supervised loss 1loss-Optimizer 1loss", | |
| "source": "Supervised loss 1", | |
| "target": "Optimizer 1", | |
| "sourceHandle": "loss", | |
| "targetHandle": "loss" | |
| }, | |
| { | |
| "id": "Graph conv 1 Repeat 3", | |
| "source": "Graph conv 1", | |
| "target": "Repeat 3", | |
| "sourceHandle": "x", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Repeat 3 Graph conv 1", | |
| "source": "Repeat 3", | |
| "target": "Graph conv 1", | |
| "sourceHandle": "output", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Dropout 1 Repeat 1", | |
| "source": "Dropout 1", | |
| "target": "Repeat 1", | |
| "sourceHandle": "x", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Repeat 1 Linear 1", | |
| "source": "Repeat 1", | |
| "target": "Linear 1", | |
| "sourceHandle": "output", | |
| "targetHandle": "x" | |
| } | |
| ] | |
| } | |