n2j.models.gnn

Various GNN models

Module Contents

Classes

GCNNet
GATNet
SageNet
GravNet

class n2j.models.gnn.GCNNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

Bases: torch.nn.Module

forward(data)

class n2j.models.gnn.GATNet(in_channels, out_channels, hidden_channels=256, kwargs={}, n_layers=3, dropout=0.0)

Bases: torch.nn.Module

forward(data)

class n2j.models.gnn.SageNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

Bases: torch.nn.Module

forward(data)

class n2j.models.gnn.GravNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

Bases: torch.nn.Module

forward(data)