Struct GatEmbeddings 

pub struct GatEmbeddings {
    pub embeddings: Vec<Vec<f64>>,
    pub config: GatConfig,
    pub num_nodes: usize,
    pub dim: usize,
}

Output embeddings from GAT inference

Fields

embeddings: Vec<Vec<f64>>

Embedding vectors indexed by node ID

config: GatConfig

Configuration used

num_nodes: usize

Number of nodes

dim: usize

Embedding dimensionality

Implementations

impl GatEmbeddings

pub fn get(&self, node: usize) -> Option<&[f64]>

Get embedding for a specific node

pub fn cosine_similarity(&self, a: usize, b: usize) -> Option<f64>

Compute cosine similarity between two nodes

pub fn top_k_similar(&self, node: usize, k: usize) -> Vec<(usize, f64)>

Get the top-k most similar nodes to a given node

Trait Implementations

impl Clone for GatEmbeddings

fn clone(&self) -> GatEmbeddings

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GatEmbeddings

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

Formats the value using the given formatter.