bioneuralnet.network_embedding.gnn_models

Classes

GAT(input_dim, hidden_dim[, layer_num, ...])
GCN(input_dim, hidden_dim[, layer_num, dropout])
GIN(input_dim, hidden_dim[, layer_num, dropout])
SAGE(input_dim, hidden_dim[, layer_num, dropout])