GNN Learner

Takes a graph object and nodes dataframe to generate a trained GNN model

Options

Hidden Channels

The number of hidden channels desired.

Hidden Layers

The number of hidden layers desired. Recommended to use a number less than the graph diameter if unsure.

Learning Rate

The learning rate to use in the optimizer

Epochs

The number of epochs to train for.

Input Ports

Icon

Pickled Graph

Icon

A table that contains nodes need to be masked and passed downstream.

Output Ports

Icon

Pytorch Geometric Model

Popular Predecessors

  • No recommendations found

Popular Successors

  • No recommendations found

Views

This node has no views

Workflows

  • No workflows found

Links

Developers

You want to see the source code for this node? Click the following button and we’ll use our super-powers to find it for you.