scirs2-graph 0.4.2

Graph processing module for SciRS2 (scirs2-graph)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! Graph Transformer models: GraphGPS and Graphormer.
//!
//! Provides two state-of-the-art graph transformer architectures:
//!
//! - **GraphGPS** (Rampasek et al. 2022): combines a local MPNN branch with a
//!   global self-attention Transformer, augmented with positional encodings.
//! - **Graphormer** (Ying et al. 2021): uses structural encodings (degree
//!   embedding, spatial encoding via SPD, virtual graph token) injected directly
//!   into the standard Transformer architecture.
//!
//! Positional encodings are provided in the [`positional_encoding`] sub-module:
//! Laplacian PE, Random-Walk PE, and all-pairs shortest paths (BFS).

pub mod graphgps;
pub mod graphormer;
pub mod positional_encoding;
pub mod types;

// Convenience re-exports
pub use graphgps::GpsModel;
pub use graphormer::GraphormerModel;
pub use positional_encoding::{all_pairs_shortest_path, laplacian_pe, rwpe};
pub use types::{
    GraphForTransformer, GraphTransformerConfig, GraphTransformerOutput, GraphormerConfig, PeType,
};