Struct GraphData 

pub struct GraphData {
    pub node_features: Vec<Vec<f64>>,
    pub adjacency: Vec<Vec<usize>>,
    pub labels: Option<Vec<usize>>,
}

Node features and graph structure for GraphSAGE

Fields

node_features: Vec<Vec<f64>>

Node feature matrix: node_features[i] = feature vector for node i

adjacency: Vec<Vec<usize>>

Adjacency list: adjacency[i] = list of neighbor indices

labels: Option<Vec<usize>>

Optional node labels for supervised training

Implementations

impl GraphData

pub fn new(features: Vec<Vec<f64>>, adjacency: Vec<Vec<usize>>) -> Result<Self>

Create a new graph from node features and adjacency list.

Validates that all adjacency indices are within bounds.

pub fn num_nodes(&self) -> usize

Number of nodes in the graph

pub fn feature_dim(&self) -> usize

Dimensionality of node features

pub fn neighbors(&self, node: usize) -> &[usize]

Get the neighbors of a node

pub fn sample_neighbors( &self, node: usize, k: usize, rng: &mut SimpleLcg, ) -> Vec<usize>

Sample up to k neighbors uniformly at random using a simple LCG PRNG

pub fn with_labels(self, labels: Vec<usize>) -> Result<Self>

Set node labels for supervised training

Trait Implementations

impl Clone for GraphData

fn clone(&self) -> GraphData

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphData

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

Formats the value using the given formatter.