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