Struct GATConfig 

pub struct GATConfig {
    pub input_dim: usize,
    pub hidden_head_dim: usize,
    pub hidden_num_heads: usize,
    pub output_head_dim: usize,
    pub output_num_heads: usize,
    pub num_layers: usize,
    pub dropout: f64,
    pub alpha: f64,
    pub concat_hidden: bool,
    pub avg_output: bool,
    pub normalize_output: bool,
    pub seed: u64,
}

Full hyperparameter configuration for a GAT model.

Fields

input_dim: usize

Dimension of raw input node features.

hidden_head_dim: usize

Per-head output dimensionality in intermediate layers.

hidden_num_heads: usize

Number of attention heads in intermediate layers.

output_head_dim: usize

Per-head output dimensionality in the final layer.

output_num_heads: usize

Number of attention heads in the final layer.

num_layers: usize

Number of stacked GAT layers (must be >= 1).

dropout: f64

Dropout rate applied to attention coefficients (0.0 = disabled).

alpha: f64

Negative slope for LeakyReLU in attention scoring.

concat_hidden: bool

If true, concatenate head outputs in intermediate layers; else average.

avg_output: bool

Average head outputs in the final layer (standard GAT for classification).

normalize_output: bool

L2-normalize final node embeddings.

seed: u64

Random seed for weight initialization.

Implementations

impl GATConfig

pub fn output_dim(&self) -> usize

Compute the total output dimensionality of this configuration.

pub fn hidden_layer_out_dim(&self) -> usize

Compute the output dimensionality of each hidden (intermediate) layer.

Trait Implementations

impl Clone for GATConfig

fn clone(&self) -> GATConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GATConfig

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

Formats the value using the given formatter.

impl Default for GATConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for GATConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for GATConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.