Struct LinkPredictor 

pub struct LinkPredictor<M: EmbeddingModel> { /* private fields */ }

Link predictor for knowledge graph completion

Implementations

impl<M: EmbeddingModel> LinkPredictor<M>

pub fn new(config: LinkPredictionConfig, model: M) -> Self

Create new link predictor

pub fn add_known_triples(&mut self, triples: &[Triple])

Add known triples for filtering

pub fn predict_tail( &self, subject: &str, predicate: &str, candidate_entities: &[String], ) -> Result<Vec<LinkPrediction>>

Predict tail entities given head and relation

pub fn predict_head( &self, predicate: &str, object: &str, candidate_entities: &[String], ) -> Result<Vec<LinkPrediction>>

Predict head entities given relation and tail

pub fn predict_relation( &self, subject: &str, object: &str, candidate_relations: &[String], ) -> Result<Vec<LinkPrediction>>

Predict relations given head and tail

pub fn predict_tails_batch( &self, queries: &[(String, String)], candidate_entities: &[String], ) -> Result<Vec<Vec<LinkPrediction>>>

Batch prediction of tails

pub fn evaluate( &self, test_triples: &[Triple], candidate_entities: &[String], ) -> Result<LinkPredictionMetrics>

Evaluate link prediction on a test set

pub fn model(&self) -> &M

Get reference to underlying model

pub fn model_mut(&mut self) -> &mut M

Get mutable reference to underlying model