dgmc.models

class dgmc.models.GIN(in_channels, out_channels, num_layers, batch_norm=False, cat=True, lin=True)[source]
forward(x, edge_index, *args)[source]
reset_parameters()[source]
class dgmc.models.SplineCNN(in_channels, out_channels, dim, num_layers, cat=True, lin=True, dropout=0.0)[source]
forward(x, edge_index, edge_attr, *args)[source]
reset_parameters()[source]
class dgmc.models.RelCNN(in_channels, out_channels, num_layers, batch_norm=False, cat=True, lin=True, dropout=0.0)[source]
forward(x, edge_index, *args)[source]
reset_parameters()[source]