jiro_nn 0.8.1

Neural Networks framework with model building & data preprocessing features.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use crate::{dataset::Dataset, datatable::DataTable};

use super::{DataTransformation, CachedConfig};

pub struct Sample {
    pub count: usize,
    pub shuffle: bool,
}

impl Sample {
    pub fn new(count: usize, shuffle: bool) -> Self {
        Self { count, shuffle }
    }
}

impl DataTransformation for Sample {
    fn transform(
        &mut self,
        _cached_config: &CachedConfig,
        dataset_config: &Dataset,
        data: &DataTable,
    ) -> (Dataset, DataTable) {
        let data = data.sample(Some(self.count), self.shuffle);
        (dataset_config.clone(), data)
    }

    fn reverse_columnswise(&mut self, data: &DataTable) -> DataTable {
        data.clone()
    }

    fn get_name(&self) -> String {
        let seed = if self.shuffle {
            rand::random::<u64>()
        } else {
            0
        };

        format!("sample({},{})", self.count, seed)
    }
}