Struct CurriculumLearningTrainer 

pub struct CurriculumLearningTrainer<M: Model> {
    pub model: M,
    pub config: CurriculumConfig,
    pub examples: Vec<CurriculumExample>,
    pub current_threshold: f32,
    pub current_epoch: usize,
    pub step_counter: usize,
    pub performance_history: Vec<f32>,
    pub difficulty_scorer: Option<DifficultyScorer>,
}

Curriculum learning trainer

Fields

model: M

The model being trained

config: CurriculumConfig

Configuration

examples: Vec<CurriculumExample>

All training examples with difficulty scores

current_threshold: f32

Current difficulty threshold

current_epoch: usize

Current epoch

step_counter: usize

Training step counter

performance_history: Vec<f32>

Performance history for adaptive curriculum

difficulty_scorer: Option<DifficultyScorer>

Difficulty scorer for dynamic difficulty estimation

Implementations

impl<M: Model<Input = Tensor, Output = Tensor>> CurriculumLearningTrainer<M>

pub fn new(model: M, config: CurriculumConfig) -> Result<Self>

Create a new curriculum learning trainer

pub fn add_examples(&mut self, examples: Vec<CurriculumExample>)

Add training examples to the curriculum

pub fn add_example(&mut self, example: CurriculumExample)

Add a single example

pub fn estimate_difficulties(&mut self) -> Result<()>

Estimate difficulty for examples without scores

pub fn get_current_curriculum(&self) -> Vec<CurriculumExample>

Get current curriculum subset

pub fn update_curriculum_threshold(&mut self) -> Result<()>

Update curriculum threshold based on strategy

pub fn train_step(&mut self) -> Result<CurriculumLearningOutput>

Train one step with curriculum

pub fn train_epoch(&mut self) -> Result<CurriculumEpochOutput>

Train for one epoch with curriculum

pub fn get_curriculum_stats(&self) -> CurriculumStats

Get curriculum statistics