Skip to main content

MetaLearningAlgorithm

trustformers_training::few_shot::meta_learning

Trait MetaLearningAlgorithm 

Source 
pub trait MetaLearningAlgorithm {
    // Required methods
    fn meta_update(
        &mut self,
        task_batch: &TaskBatch,
    ) -> Result<MetaUpdateResult>;
    fn adapt(
        &self,
        support_set: &TaskData,
        adaptation_steps: usize,
    ) -> Result<ModelParameters>;
    fn evaluate(
        &self,
        params: &ModelParameters,
        query_set: &TaskData,
    ) -> Result<f32>;
}
Expand description

Meta-learning algorithm trait

Required Methods§

Source

fn meta_update(&mut self, task_batch: &TaskBatch) -> Result<MetaUpdateResult>

Source

fn adapt( &self, support_set: &TaskData, adaptation_steps: usize, ) -> Result<ModelParameters>

Source

fn evaluate( &self, params: &ModelParameters, query_set: &TaskData, ) -> Result<f32>

Implementors§