Struct NodeEmbeddings 

pub struct NodeEmbeddings {
    pub node_names: Vec<String>,
    pub embeddings: Vec<Vec<f64>>,
    pub dim: usize,
}

Container for node feature embeddings

Fields

node_names: Vec<String>

Node index to name mapping (optional)

embeddings: Vec<Vec<f64>>

Feature matrix (n_nodes × embedding_dim)

dim: usize

Embedding dimensionality

Implementations

impl NodeEmbeddings

pub fn new(embeddings: Vec<Vec<f64>>) -> Result<Self>

Create embeddings from raw feature matrix

pub fn random(n_nodes: usize, dim: usize) -> Self

Create random initial embeddings

pub fn one_hot(n_nodes: usize) -> Self

Create one-hot embeddings

pub fn n_nodes(&self) -> usize

Get the number of nodes

pub fn get(&self, i: usize) -> Option<&Vec<f64>>

Get embedding for node i

pub fn apply_layer<L: MessagePassingLayer>( &self, layer: &L, adjacency: &[(usize, usize, f64)], ) -> Result<NodeEmbeddings>

Apply a GNN layer to these embeddings

Trait Implementations

impl Clone for NodeEmbeddings

fn clone(&self) -> NodeEmbeddings

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NodeEmbeddings

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

Formats the value using the given formatter.