Struct GnnLayer 

pub struct GnnLayer { /* private fields */ }

A simplified GNN layer for embedding transformation.

This implements a message-passing mechanism where node embeddings are updated based on their neighbors, similar to the ruvector GNN layer.

Implementations

impl GnnLayer

pub fn new(input_dim: usize, hidden_dim: usize, num_heads: usize) -> Self

Create a new GNN layer.

pub fn forward( &self, node_embedding: &[f32], neighbor_embeddings: &[Vec<f32>], edge_weights: &[f32], ) -> Vec<f32>

Forward pass through the GNN layer.

pub fn update(&mut self, query: &[f32], target_score: f32, learning_rate: f32)

Update weights based on feedback.