Struct BatchSampler 

pub struct BatchSampler { /* private fields */ }

A sampler that creates batches of data with a specified batch size.

This struct is used to divide data into batches for processing in the OCR pipeline. It implements the Sampler trait for String data.

Implementations

impl BatchSampler

pub fn new(batch_size: usize) -> Self

Creates a new BatchSampler with the specified batch size.

Arguments

• batch_size - The size of each batch.

Returns

A new BatchSampler instance.

pub fn batch_size(&self) -> usize

Returns the batch size.

Returns

The batch size.

pub fn batches<'a, T>(&self, data: &'a [T]) -> impl Iterator<Item = &'a [T]>

Creates an iterator over batches of data.

Arguments

• data - A slice of data to be batched.

Returns

An iterator over batches of data.

pub fn batches_with_indexes<'a, T>( &self, data: &'a [T], ) -> impl Iterator<Item = (&'a [T], Vec<usize>)>

Creates an iterator over batches of data with their indexes.

Arguments

• data - A slice of data to be batched.

Returns

An iterator over tuples containing batches of data and their indexes.

pub fn sample_batch(&self, data: Vec<String>) -> Vec<BatchData>

Samples batches of data from a vector of strings.

Arguments

• data - A vector of strings to be batched.

Returns

A vector of BatchData instances.

Trait Implementations

impl Debug for BatchSampler

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

Formats the value using the given formatter.

impl Sampler<String> for BatchSampler

fn sample(&self, data: Vec<String>) -> Vec<Self::BatchData>

Samples batches of data from a vector of strings.

This method implements the Sampler trait for String data.

Arguments

• data - A vector of strings to be batched.

Returns

A vector of BatchData instances.

type BatchData = BatchData

The type of batch data produced by this sampler.

fn sample_slice(&self, data: &[T]) -> Vec<Self::BatchData>

where T: Clone,

Samples the given slice of data into batches.

fn sample_iter<I>(&self, data: I) -> Vec<Self::BatchData>

where I: IntoIterator<Item = T>,

Samples the given iterator of data into batches.