Skip to main content

DataAugmenter

tensorlogic_train

Trait DataAugmenter 

Source 
pub trait DataAugmenter {
    // Required method
    fn augment(
        &self,
        data: &ArrayView2<'_, f64>,
        rng: &mut StdRng,
    ) -> TrainResult<Array2<f64>>;
}
Expand description

Trait for data augmentation strategies.

Required Methods§

Source

fn augment( &self, data: &ArrayView2<'_, f64>, rng: &mut StdRng, ) -> TrainResult<Array2<f64>>

Augment the given data.

§Arguments
  • data - Input data to augment
  • rng - Random number generator for stochastic augmentation
§Returns

Augmented data with the same shape as input

Implementations on Foreign Types§

Source§

impl DataAugmenter for Box<dyn AugmenterClone>

Source§

fn augment( &self, data: &ArrayView2<'_, f64>, rng: &mut StdRng, ) -> TrainResult<Array2<f64>>

Implementors§