Struct GraphMAEConfig 

pub struct GraphMAEConfig {
    pub mask_ratio: f32,
    pub num_layers: usize,
    pub hidden_dim: usize,
    pub num_heads: usize,
    pub decoder_layers: usize,
    pub re_mask_ratio: f32,
    pub loss_fn: LossFn,
    pub input_dim: usize,
}

Configuration for a GraphMAE model.

Fields

mask_ratio: f32

Fraction of nodes to mask (default 0.5).

num_layers: usize

Number of GAT encoder layers.

hidden_dim: usize

Hidden / latent dimension.

num_heads: usize

Number of attention heads per encoder layer.

decoder_layers: usize

Number of decoder layers.

re_mask_ratio: f32

Secondary mask ratio applied to latent before decoding (default 0.0).

loss_fn: LossFn

Reconstruction loss function.

input_dim: usize

Input feature dimension.

Trait Implementations

impl Clone for GraphMAEConfig

fn clone(&self) -> GraphMAEConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GraphMAEConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for GraphMAEConfig

fn default() -> Self

Returns the “default value” for a type.