Struct GraphSAGEEmbeddings 

pub struct GraphSAGEEmbeddings {
    pub embeddings: Vec<Vec<f64>>,
    pub num_nodes: usize,
    pub dim: usize,
}

Node embeddings produced by GraphSAGEModel.

Fields

embeddings: Vec<Vec<f64>>

num_nodes: usize

dim: usize

Implementations

impl GraphSAGEEmbeddings

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

Get embedding for node v.

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

Cosine similarity between nodes a and b. Returns None if either embedding is zero or out of bounds.

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

Top-k nodes most similar to query_node (excludes query_node itself).

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

Build a map from node index to label embedding pairs.

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.