Struct GraphSageEmbeddings 

pub struct GraphSageEmbeddings {
    pub embeddings: Vec<Vec<f64>>,
    pub config: GraphSageConfig,
    pub num_nodes: usize,
    pub dim: usize,
}

Output embeddings from GraphSAGE inference

Fields

embeddings: Vec<Vec<f64>>

Embedding vectors indexed by node ID

config: GraphSageConfig

Configuration used for generation

num_nodes: usize

Number of nodes

dim: usize

Embedding dimensionality

Implementations

impl GraphSageEmbeddings

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

Get the embedding for a specific node

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

Compute cosine similarity between two node embeddings

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

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

Returns a sorted list of (node_id, similarity_score) pairs.

pub fn labeled_embeddings( &self, labels: &[usize], ) -> HashMap<usize, Vec<Vec<f64>>>

Build a lookup from node label to embedding for classified nodes

Trait Implementations

impl Clone for GraphSageEmbeddings

fn clone(&self) -> GraphSageEmbeddings

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphSageEmbeddings

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

Formats the value using the given formatter.