Struct GraphNeuralEngine 

pub struct GraphNeuralEngine { /* private fields */ }

Graph Neural Network engine

Implementations

impl GraphNeuralEngine

pub fn new(config: GnnConfig) -> Self

Create a new GNN engine

pub fn load_model(&mut self, _model_path: &str) -> Result<()>

Load pre-trained model weights

pub fn classify_node( &self, node_id: &NodeId, _features: &[f32], ) -> Result<NodeClassification>

Classify a node based on its features and neighbors

pub fn predict_link( &self, node1: &NodeId, node2: &NodeId, ) -> Result<LinkPrediction>

Predict likelihood of a link between two nodes

pub fn embed_graph(&self, node_ids: &[NodeId]) -> Result<GraphEmbedding>

Generate embedding for entire graph or subgraph

pub fn update_embeddings( &mut self, graph_structure: &GraphStructure, ) -> Result<()>

Update node embeddings using message passing

pub fn get_node_embedding(&self, node_id: &NodeId) -> Option<&Vec<f32>>

Get embedding for a specific node

pub fn classify_nodes_batch( &self, nodes: &[(NodeId, Vec<f32>)], ) -> Result<Vec<NodeClassification>>

Batch node classification

pub fn predict_links_batch( &self, pairs: &[(NodeId, NodeId)], ) -> Result<Vec<LinkPrediction>>

Batch link prediction