OA LibraryReference

oa::data::DataLoader

Stateful batch iterator over a borrowed Dataset. Corresponds to oa::DataLoader.

StRust public API

Declaration

pub struct DataLoader<'ds> { /* private fields */ }

Public Rust declaration.

Public Methods

pub fn next_batch(&mut self) -> Result<Option<Batch>>

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

pub fn num_batches(&self) -> i64

pub fn current_batch(&self) -> i64

Associated Functions

pub fn new(dataset: &'ds dyn Dataset, config: DataLoaderConfig) -> Result<Self>

Public Method Documentation

pub fn next_batch(
	&mut self
) -> Result<Option<Batch>>

No public source comment is attached to this declaration.

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

No public source comment is attached to this declaration.

pub fn num_batches(
	&self
) -> i64

No public source comment is attached to this declaration.

pub fn current_batch(
	&self
) -> i64

No public source comment is attached to this declaration.

pub fn new(
	dataset: &'ds dyn Dataset,
	config: DataLoaderConfig
) -> Result<Self>

No public source comment is attached to this declaration.