Struct TrainConfig 

pub struct TrainConfig {
    pub epochs: usize,
    pub batch_size: usize,
    pub shuffle_seed: Option<u64>,
    /* private fields */
}

Fields

epochs: usize

batch_size: usize

shuffle_seed: Option<u64>

Implementations

impl TrainConfig

pub fn new<O: Optimizer + 'static>(optimizer: O) -> Self

pub fn sgd(lr: Float) -> Self

pub fn adam(lr: Float) -> Self

pub fn epochs(self, epochs: usize) -> Self

pub fn batch_size(self, batch_size: usize) -> Self

pub fn shuffle_seed(self, shuffle_seed: u64) -> Self

Trait Implementations

impl Debug for TrainConfig

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

Formats the value using the given formatter.

impl Default for TrainConfig

fn default() -> Self

Returns the “default value” for a type.