Struct EnsembleEmbedder 

pub struct EnsembleEmbedder { /* private fields */ }

Implementations

impl EnsembleEmbedder

pub fn new( models: Vec<Box<dyn Fn(&str) -> Vec<f64> + Send + Sync>>, config: EnsembleConfig, ) -> Result<Self>

Create a new ensemble embedder.

Errors

Returns an error if models is empty.

pub fn num_models(&self) -> usize

Return the number of models in the ensemble.

pub fn set_weights(&mut self, weights: Vec<f64>) -> Result<()>

Set per-model weights for WeightedAverage.

Weights are automatically normalised to sum to 1.

Errors

Returns an error if the weight vector length does not match the model count, or if any weight is negative or not finite, or if the sum is zero.

pub fn embed(&self, key: &str) -> Result<Vec<f64>>

Produce an ensemble embedding for the given entity key.

Errors

• Dimensionality mismatch between model output and config.output_dim.
• Stacking strategy called before EnsembleEmbedder::fit_stacking.

pub fn fit_stacking( &mut self, validation_pairs: &[(&str, Vec<f64>)], ) -> Result<()>

Train the stacking meta-learner on a validation set.

validation_pairs is a slice of (key, target_embedding) pairs where target_embedding is the ground-truth embedding (e.g. from a stronger reference model or link-prediction evaluation).

Errors

Returns an error if the validation set is empty, or if any target embedding has the wrong dimensionality.

pub fn eval_cosine( &self, reference: &impl Fn(&str) -> Vec<f64>, validation_keys: &[&str], ) -> Result<f64>

Compute the average cosine similarity between ensemble output and a reference model on the given validation keys. Useful for tuning weights.

pub fn derive_weights( &mut self, reference: &impl Fn(&str) -> Vec<f64>, validation_keys: &[&str], ) -> Result<()>

Derive performance-based weights from cosine similarity scores on a validation set.

Each model’s weight is proportional to its mean cosine similarity to a reference embedding. Models with zero weight (similarity ≤ 0) are assigned a small ε = 1e-6 to keep them in the ensemble.

Trait Implementations

impl Debug for EnsembleEmbedder

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

Formats the value using the given formatter.