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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

use std::{fs::File, io::Write};

use serde::{Deserialize, Serialize};

use crate::linalg::Scalar;

#[derive(Clone, Serialize, Deserialize, Debug, PartialEq)]
pub struct ModelEvaluation {
    pub folds: Vec<TrainingEvaluation>,
}

#[derive(Clone, Serialize, Deserialize, Debug, PartialEq)]
pub struct TrainingEvaluation {
    pub epochs: Vec<EpochEvaluation>,
}

#[derive(Clone, Serialize, Deserialize, Debug, PartialEq)]
pub struct EpochEvaluation {
    pub train_loss: Scalar,
    pub test_loss_avg: Scalar,
    pub test_loss_std: Scalar,
    pub r2: Scalar,
}

impl ModelEvaluation {
    pub fn new_empty() -> Self {
        Self { folds: vec![] }
    }

    pub fn add_fold(&mut self, fold: TrainingEvaluation) {
        self.folds.push(fold);
    }

    pub fn epochs_avg_train_loss(&self) -> Vec<Scalar> {
        let mut avg = vec![0.0; self.folds[0].epochs.len()];
        for fold in &self.folds {
            for (i, epoch) in fold.epochs.iter().enumerate() {
                avg[i] += epoch.train_loss;
            }
        }
        for i in 0..avg.len() {
            avg[i] /= self.folds.len() as Scalar;
        }
        avg
    }

    pub fn epochs_std_train_loss(&self) -> Vec<Scalar> {
        let avg = self.epochs_avg_train_loss();
        let mut std = vec![0.0; self.folds[0].epochs.len()];
        for fold in &self.folds {
            for (i, epoch) in fold.epochs.iter().enumerate() {
                std[i] += (epoch.train_loss - avg[i]).powi(2);
            }
        }
        for i in 0..std.len() {
            std[i] = (std[i] / self.folds.len() as Scalar).sqrt();
        }
        std
    }

    pub fn epochs_avg_test_loss(&self) -> Vec<Scalar> {
        let mut avg = vec![0.0; self.folds[0].epochs.len()];
        for fold in &self.folds {
            for (i, epoch) in fold.epochs.iter().enumerate() {
                avg[i] += epoch.test_loss_avg;
            }
        }
        for i in 0..avg.len() {
            avg[i] /= self.folds.len() as Scalar;
        }
        avg
    }

    pub fn epochs_std_test_loss(&self) -> Vec<Scalar> {
        let avg = self.epochs_avg_test_loss();
        let mut std = vec![0.0; self.folds[0].epochs.len()];
        for fold in &self.folds {
            for (i, epoch) in fold.epochs.iter().enumerate() {
                std[i] += (epoch.test_loss_avg - avg[i]).powi(2);
            }
        }
        for i in 0..std.len() {
            std[i] = (std[i] / self.folds.len() as Scalar).sqrt();
        }
        std
    }

    pub fn from_json_file<S: AsRef<str>>(path: S) -> Self {
        let file = File::open(path.as_ref()).unwrap();
        serde_json::from_reader(file).unwrap()
    }

    pub fn to_json_file<S: AsRef<str>>(&self, path: S) {
        let mut file = File::create(path.as_ref()).unwrap();
        let json_string = serde_json::to_string_pretty(self).unwrap();
        file.write_all(json_string.as_bytes()).unwrap();
    }

    pub fn get_n_epochs(&self) -> usize {
        self.folds[0].epochs.len()
    }

    pub fn get_n_folds(&self) -> usize {
        self.folds.len()
    }
}

impl TrainingEvaluation {
    pub fn new_empty() -> Self {
        Self { epochs: vec![] }
    }

    pub fn add_epoch(&mut self, epoch: EpochEvaluation) {
        self.epochs.push(epoch);
    }

    pub fn get_final_epoch(&self) -> EpochEvaluation {
        self.epochs[self.epochs.len() - 1].clone()
    }

    pub fn get_final_test_loss_avg(&self) -> Scalar {
        self.get_final_epoch().test_loss_avg
    }

    pub fn get_final_test_loss_std(&self) -> Scalar {
        self.get_final_epoch().test_loss_std
    }

    pub fn get_final_r2(&self) -> Scalar {
        self.get_final_epoch().r2
    }
}

impl EpochEvaluation {
    pub fn new(
        train_loss: Scalar,
        test_loss_avg: Scalar,
        test_loss_std: Scalar,
        r2: Scalar,
    ) -> Self {
        Self {
            train_loss,
            test_loss_avg,
            test_loss_std,
            r2,
        }
    }
}