Struct AugmentationPipeline

pub struct AugmentationPipeline {
    pub name: String,
    /* private fields */
}

A pipeline of augmenters

Executes many augmenters at once

Example

 use fraug::Dataset;
 use fraug::augmenters::*;

 let series = vec![1.0; 100];
 let mut set = Dataset {
    features: vec![series],
    labels: vec![String::from("1")],
 };

 let pipeline = AugmentationPipeline::new()
                + Repeat::new(5)
                + Crop::new(20)
                + Jittering::new(0.2);

 pipeline.augment_batch(&mut set, true, false);

 assert_eq!(set.features.len(), 5);
 assert_eq!(set.features[3].len(), 20);

Fields

name: String

Implementations

impl AugmentationPipeline

pub fn new() -> Self

Creates an empty pipeline

pub fn add<T: Augmenter + 'static + Sync>(&mut self, augmenter: T)

Add an augmenter to the pipeline

Has the same effect as using the + operator

Trait Implementations

impl<T: Augmenter + 'static + Sync> Add<T> for AugmentationPipeline

type Output = AugmentationPipeline

The resulting type after applying the + operator.

fn add(self, rhs: T) -> Self::Output

Performs the + operation.

impl Augmenter for AugmentationPipeline

fn augment_batch(&self, input: &mut Dataset, parallel: bool, per_sample: bool)

Augment a whole batch

fn augment_one(&self, x: &[f64]) -> Vec<f64>

Augment one time series

fn get_probability(&self) -> f64

Get the probability that this augmenter will augment a series in a batch

fn set_probability(&mut self, probability: f64)

By setting a probability with this function the augmenter will only augment a series in a batch with the specified probability

fn get_name(&self) -> String

fn supports_per_sample(&self) -> bool

Indicate whether this augmenter supports per-sample chaining. By default, return true. Augmenters that need a batch level view should override this to return false.