Struct DiffusionEmbeddingModel 

pub struct DiffusionEmbeddingModel { /* private fields */ }

Main diffusion embedding model

Implementations

impl DiffusionEmbeddingModel

pub fn new(config: ModelConfig, diffusion_config: DiffusionConfig) -> Self

Create new diffusion embedding model

pub fn generate_embeddings( &self, condition: Option<&Array2<f64>>, num_samples: usize, guidance_scale: f64, ) -> Result<Array2<f64>>

Generate embeddings using diffusion sampling

pub fn generate_conditional_embeddings( &self, entity_types: &[String], relation_types: &[String], ) -> Result<(Array2<f64>, Array2<f64>)>

Generate conditional embeddings for specific entities/relations

pub fn interpolate_embeddings( &self, embedding1: &Array2<f64>, embedding2: &Array2<f64>, alpha: f64, ) -> Result<Array2<f64>>

Interpolate between embeddings

pub fn edit_embedding( &self, original: &Array2<f64>, edit_direction: &Array2<f64>, strength: f64, ) -> Result<Array2<f64>>

Edit embedding with diffusion inversion

Trait Implementations

impl Clone for DiffusionEmbeddingModel

fn clone(&self) -> DiffusionEmbeddingModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DiffusionEmbeddingModel

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

Formats the value using the given formatter.

impl EmbeddingModel for DiffusionEmbeddingModel

fn config(&self) -> &ModelConfig

fn model_id(&self) -> &Uuid

fn model_type(&self) -> &'static str

fn add_triple(&mut self, triple: Triple) -> Result<()>

fn train<'life0, 'async_trait>( &'life0 mut self, epochs: Option<usize>, ) -> Pin<Box<dyn Future<Output = Result<TrainingStats>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn get_entity_embedding(&self, entity: &str) -> Result<Vector>

fn get_relation_embedding(&self, relation: &str) -> Result<Vector>

fn score_triple( &self, subject: &str, predicate: &str, object: &str, ) -> Result<f64>

fn predict_objects( &self, subject: &str, predicate: &str, k: usize, ) -> Result<Vec<(String, f64)>>

fn predict_subjects( &self, predicate: &str, object: &str, k: usize, ) -> Result<Vec<(String, f64)>>

fn predict_relations( &self, subject: &str, object: &str, k: usize, ) -> Result<Vec<(String, f64)>>

fn get_entities(&self) -> Vec<String>

fn get_relations(&self) -> Vec<String>

fn get_stats(&self) -> ModelStats

fn save(&self, _path: &str) -> Result<()>

fn load(&mut self, _path: &str) -> Result<()>

fn clear(&mut self)

fn is_trained(&self) -> bool

fn encode<'life0, 'life1, 'async_trait>( &'life0 self, texts: &'life1 [String], ) -> Pin<Box<dyn Future<Output = Result<Vec<Vec<f32>>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Encode text strings into embeddings