Struct GraphSageEncoder 

pub struct GraphSageEncoder { /* private fields */ }

GraphSAGE encoder.

Implementations

impl GraphSageEncoder

pub fn new(config: &GraphSageConfig) -> GnnResult<Self>

Create a new encoder with seed 42.

pub fn new_with_seed(config: &GraphSageConfig, seed: u64) -> GnnResult<Self>

Create a new encoder with an explicit seed for reproducibility.

pub fn encode(&self, graph: &KgGraph) -> GnnResult<EntityEmbeddings>

Encode all nodes in graph and return entity embeddings.

A fresh RNG derived from the stored seed is created on each call, so repeated calls on the same encoder produce identical results.

pub fn train( &mut self, graph: &KgGraph, num_epochs: usize, ) -> GnnResult<TrainingHistory>

Train the encoder for num_epochs using link-prediction loss.

pub fn compute_grad_and_param_for_test(&mut self, graph: &KgGraph) -> (f64, f64)

Return (analytic_grad_of_W1[0][0], W1[0][0]) for a fixed triple (0 → 1) as positive pair. Intended for finite-difference checks.

pub fn compute_loss_with_perturb(&mut self, graph: &KgGraph, eps: f64) -> f64

Compute loss with W1[0][0] perturbed by eps, then restore. Uses the same RNG seed as compute_grad_and_param_for_test.