rotta_rs 0.0.5

a Deep Learning library with rust language
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

use std::sync::{ Arc, Mutex };

use rand::{ rngs::StdRng, SeedableRng };

use crate::rotta_rs_module::{ NodeType, WeightInitialization };

pub struct Module {
    pub parameters: Arc<Mutex<Vec<NodeType>>>,
    pub initialization: WeightInitialization,
    pub eval_handlers: Vec<Arc<Mutex<bool>>>,

    // rng
    pub rng: StdRng,
}

impl Module {
    // initialization
    pub fn init() -> Module {
        Module {
            parameters: Arc::new(Mutex::new(Vec::new())),
            initialization: WeightInitialization::He,
            eval_handlers: vec![],

            // rng
            rng: StdRng::seed_from_u64(42),
        }
    }

    // parameters
    pub fn parameters(&self) -> Arc<Mutex<Vec<Arc<Mutex<crate::rotta_rs_module::Node>>>>> {
        self.parameters.clone()
    }

    // eval & training
    pub fn train(&self) {
        for p in &self.eval_handlers {
            *p.lock().unwrap() = false;
        }
    }

    pub fn eval(&self) {
        for p in &self.eval_handlers {
            *p.lock().unwrap() = true;
        }
    }

    // update seed
    pub fn update_seed(&mut self, seed: u64) {
        self.rng = StdRng::seed_from_u64(seed);
    }

    pub fn update_initialization(&mut self, initialization: WeightInitialization) {
        self.initialization = initialization;
    }
}