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

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

use super::{DataTransformation, CachedConfig};

pub struct AttachIds(pub String);

impl AttachIds {
    pub fn new(id_column_name: &str) -> Self {
        Self(id_column_name.to_string())
    }
}

impl DataTransformation for AttachIds {
    fn transform(
        &mut self,
        _cached_config: &CachedConfig,
        dataset_config: &Dataset,
        data: &DataTable,
    ) -> (Dataset, DataTable) {
        let mut feature = Feature::default();
        feature.name = self.0.clone();
        feature.used_in_model = true;
        feature.is_id = true;
        let configuration = dataset_config.with_added_feature(feature);
        let data = data.with_autoincrement_id_column(&self.0.clone());
        (configuration, data)
    }

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

    fn get_name(&self) -> String {
        format!("attach_ids({})", self.0)
    }
}