Struct BatchConfig 

pub struct BatchConfig {
    pub batch_size: usize,
    pub drop_last: bool,
    pub sampling_strategy: SamplingStrategy,
    pub padding_strategy: PaddingStrategy,
    pub collation_strategy: CollationStrategy,
    pub max_sequence_length: Option<usize>,
    pub padding_value: f32,
    pub seed: Option<u64>,
}

Batch configuration

Fields

batch_size: usize

Batch size

drop_last: bool

Whether to drop the last incomplete batch

sampling_strategy: SamplingStrategy

Sampling strategy

padding_strategy: PaddingStrategy

Padding strategy (for sequences)

collation_strategy: CollationStrategy

Collation strategy

max_sequence_length: Option<usize>

Maximum sequence length (for padding)

padding_value: f32

Padding value

seed: Option<u64>

Random seed for reproducibility

Implementations

impl BatchConfig

pub fn new(batch_size: usize) -> Self

Create new batch configuration

pub fn with_drop_last(self, drop_last: bool) -> Self

Set whether to drop last incomplete batch

pub fn with_sampling_strategy(self, strategy: SamplingStrategy) -> Self

Set sampling strategy

pub fn with_padding_strategy(self, strategy: PaddingStrategy) -> Self

Set padding strategy

pub fn with_collation_strategy(self, strategy: CollationStrategy) -> Self

Set collation strategy

pub fn with_max_sequence_length(self, max_len: usize) -> Self

Set maximum sequence length

pub fn with_padding_value(self, value: f32) -> Self

Set padding value

pub fn with_seed(self, seed: u64) -> Self

Set random seed

Trait Implementations

impl Clone for BatchConfig

fn clone(&self) -> BatchConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BatchConfig

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

Formats the value using the given formatter.

impl Default for BatchConfig

fn default() -> Self

Returns the “default value” for a type.