astrai 2.2.0

A pretty bad neural network library
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

use super::*;

impl Network {
	pub fn get_params_list(&self) -> Vec<f64> {
        let mut params = Vec::new();

        for connection in self.connections.iter() {
            params.push(connection.weight);
        }

        for layer in self.layers.iter() {
            for bias in layer.bias.iter() {
                params.push(*bias);
            }
        }

        params
    }

    pub fn apply_params(&mut self, params: Vec<f64>) {
        let mut param_idx = 0;

        for connection in self.connections.iter_mut() {
            connection.weight = params[param_idx];
            param_idx += 1;
        }

        for layer in self.layers.iter_mut() {
            for bias in layer.bias.iter_mut() {
                *bias = params[param_idx];
                param_idx += 1;
            }
        }
    }
}