pub struct ContinualLearningModel {Show 21 fields
pub config: ContinualLearningConfig,
pub model_id: Uuid,
pub embeddings: Array2<f32>,
pub task_specific_embeddings: HashMap<String, Array2<f32>>,
pub episodic_memory: VecDeque<MemoryEntry>,
pub semantic_memory: HashMap<String, Array1<f32>>,
pub ewc_states: Vec<EWCState>,
pub synaptic_importance: Array2<f32>,
pub parameter_trajectory: Array2<f32>,
pub current_task: Option<TaskInfo>,
pub task_history: Vec<TaskInfo>,
pub task_boundaries: Vec<usize>,
pub network_columns: Vec<Array2<f32>>,
pub lateral_connections: Vec<Array2<f32>>,
pub generator: Option<Array2<f32>>,
pub discriminator: Option<Array2<f32>>,
pub entities: HashMap<String, usize>,
pub relations: HashMap<String, usize>,
pub examples_seen: usize,
pub training_stats: Option<TrainingStats>,
pub is_trained: bool,
}Fields§
§config: ContinualLearningConfig§model_id: Uuid§embeddings: Array2<f32>§task_specific_embeddings: HashMap<String, Array2<f32>>§episodic_memory: VecDeque<MemoryEntry>§semantic_memory: HashMap<String, Array1<f32>>§ewc_states: Vec<EWCState>§synaptic_importance: Array2<f32>§parameter_trajectory: Array2<f32>§current_task: Option<TaskInfo>§task_history: Vec<TaskInfo>§task_boundaries: Vec<usize>§network_columns: Vec<Array2<f32>>§lateral_connections: Vec<Array2<f32>>§generator: Option<Array2<f32>>§discriminator: Option<Array2<f32>>§entities: HashMap<String, usize>§relations: HashMap<String, usize>§examples_seen: usize§training_stats: Option<TrainingStats>§is_trained: boolImplementations§
Source§impl ContinualLearningModel
impl ContinualLearningModel
pub fn new(config: ContinualLearningConfig) -> Self
pub fn start_task(&mut self, task_id: String, task_type: String) -> Result<()>
pub async fn add_example( &mut self, data: Array1<f32>, target: Array1<f32>, task_id: Option<String>, ) -> Result<()>
pub fn get_task_performance(&self) -> HashMap<String, f32>
pub fn evaluate_forgetting(&self) -> f32
Trait Implementations§
Source§impl Debug for ContinualLearningModel
impl Debug for ContinualLearningModel
Source§impl EmbeddingModel for ContinualLearningModel
impl EmbeddingModel for ContinualLearningModel
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
Auto Trait Implementations§
impl Freeze for ContinualLearningModel
impl RefUnwindSafe for ContinualLearningModel
impl Send for ContinualLearningModel
impl Sync for ContinualLearningModel
impl Unpin for ContinualLearningModel
impl UnsafeUnpin for ContinualLearningModel
impl UnwindSafe for ContinualLearningModel
Blanket Implementations§
Source§impl<'a, T, E> AsTaggedExplicit<'a, E> for Twhere
T: 'a,
impl<'a, T, E> AsTaggedExplicit<'a, E> for Twhere
T: 'a,
Source§impl<'a, T, E> AsTaggedImplicit<'a, E> for Twhere
T: 'a,
impl<'a, T, E> AsTaggedImplicit<'a, E> for Twhere
T: 'a,
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> Read<Exclusive, BecauseExclusive> for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.