DataLoader

quantrs2_ml::pytorch_api

Trait DataLoader 

Source 
pub trait DataLoader {
    // Required methods
    fn next_batch(&mut self) -> Result<Option<(SciRS2Array, SciRS2Array)>>;
    fn reset(&mut self);
    fn batch_size(&self) -> usize;
}
Expand description

Data loader trait

Required Methods§

Source

fn next_batch(&mut self) -> Result<Option<(SciRS2Array, SciRS2Array)>>

Get next batch

Source

fn reset(&mut self)

Reset to beginning

Source

fn batch_size(&self) -> usize

Get batch size

Implementors§