torsh-graph 0.1.3

Graph neural network components for ToRSh - powered by SciRS2
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
//! Graph convolution layers

pub mod gat;
pub mod gcn;
pub mod gin;
pub mod heterogeneous;
pub mod mpnn;
pub mod sage;
pub mod transformer;

pub use gat::GATConv;
pub use gcn::GCNConv;
pub use gin::GINConv;
pub use heterogeneous::{EdgeType, HeteroGNN, HeteroGraphData, NodeType};
pub use mpnn::{AggregationType, MPNNConv};
pub use sage::SAGEConv;
pub use transformer::GraphTransformer;