Struct BaseModel 

pub struct BaseModel {
    pub config: ModelConfig,
    pub model_id: Uuid,
    pub entity_to_id: HashMap<String, usize>,
    pub id_to_entity: HashMap<usize, String>,
    pub relation_to_id: HashMap<String, usize>,
    pub id_to_relation: HashMap<usize, String>,
    pub triples: Vec<(usize, usize, usize)>,
    pub positive_triples: HashSet<(usize, usize, usize)>,
    pub is_trained: bool,
    pub creation_time: DateTime<Utc>,
    pub last_training_time: Option<DateTime<Utc>>,
}

Core data structures and functionality shared by all embedding models

Fields

config: ModelConfig

Model configuration

model_id: Uuid

Unique model identifier

entity_to_id: HashMap<String, usize>

Entity to index mapping

id_to_entity: HashMap<usize, String>

Index to entity mapping

relation_to_id: HashMap<String, usize>

Relation to index mapping

id_to_relation: HashMap<usize, String>

Index to relation mapping

triples: Vec<(usize, usize, usize)>

Training triples (subject_id, predicate_id, object_id)

positive_triples: HashSet<(usize, usize, usize)>

Set of all positive triples for fast lookup

is_trained: bool

Whether the model has been trained

creation_time: DateTime<Utc>

Model creation time

last_training_time: Option<DateTime<Utc>>

Last training time

Implementations

impl BaseModel

pub fn new(config: ModelConfig) -> Self

Create a new base model

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

Add a triple to the model

pub fn get_entity_id(&self, entity: &str) -> Option<usize>

Get entity ID

pub fn get_relation_id(&self, relation: &str) -> Option<usize>

Get relation ID

pub fn get_entity(&self, id: usize) -> Option<&String>

Get entity string from ID

pub fn get_relation(&self, id: usize) -> Option<&String>

Get relation string from ID

pub fn num_entities(&self) -> usize

Get number of entities

pub fn num_relations(&self) -> usize

Get number of relations

pub fn num_triples(&self) -> usize

Get number of triples

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

Get all entity strings

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

Get all relation strings

pub fn has_triple( &self, subject_id: usize, predicate_id: usize, object_id: usize, ) -> bool

Check if a triple exists in the knowledge base

pub fn generate_negative_samples<R>( &self, num_samples: usize, rng: &mut Random<R>, ) -> Vec<(usize, usize, usize)>

where R: RngCore,

Generate negative samples for training

pub fn get_stats(&self, model_type: &str) -> ModelStats

Get model statistics

pub fn clear(&mut self)

Clear all data

pub fn mark_trained(&mut self)

Mark model as trained

Trait Implementations

impl Clone for BaseModel

fn clone(&self) -> BaseModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BaseModel

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

Formats the value using the given formatter.