Struct DataLoader 

pub struct DataLoader {
    pub batch_size: usize,
    pub shuffle: bool,
    pub file_path: Option<String>,
}

Utility for batching and streaming sentence data.

Fields

batch_size: usize

shuffle: bool

file_path: Option<String>

Implementations

impl DataLoader

pub fn new(batch_size: usize, shuffle: bool) -> Self

Creates a new data loader with the given batch size and options.

pub fn set_file_path(&mut self, path: String)

Sets the file path for lazy loading.

pub fn load_batches(&self, sentences: &[Vec<String>]) -> Vec<Vec<Vec<String>>>

Groups sentences into fixed-size batches.

pub fn load_lazily(&self, file_path: &str) -> Result<Vec<Vec<String>>, String>

Loads sentences from a file.

pub fn stream_sentences( &self, file_path: &str, ) -> Result<Box<dyn Iterator<Item = Vec<String>>>, String>

Returns a lazy iterator over sentences from a file.

Trait Implementations

impl Clone for DataLoader

fn clone(&self) -> DataLoader

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DataLoader

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.