gammagl.models.JKNet

class JKNet(dataset, mode='max', num_layers=6, hidden=16, drop=0.5)[source]
forward(x, edge_index, edge_weight, num_nodes)[source]