Struct GraphMAE 

pub struct GraphMAE { /* private fields */ }

GraphMAE self-supervised model.

Implementations

impl GraphMAE

pub fn new(config: GraphMAEConfig) -> Result<Self, GnnError>

Construct a new GraphMAE model from configuration.

Errors

Returns GnnError::LayerConfig if dimensions are incompatible.

pub fn train_step(&self, graph: &GraphData) -> f32

Run one training step: mask -> encode -> re-mask -> decode -> loss. Returns the reconstruction loss computed only on masked nodes.

pub fn encode(&self, graph: &GraphData) -> Vec<Vec<f32>>

Encode without masking (inference mode). Returns latent embeddings for all nodes.

pub fn get_embeddings(&self, graph: &GraphData) -> Vec<Vec<f32>>

Returns node-level representations for downstream tasks.