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

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

pub use dropout::*;
use crate::rotta_rs_module::Module;
use rand::distr::Bernoulli;

impl Module {
    pub fn dropout_init(&mut self, p: f64) -> Dropout {
        let bernouli = Bernoulli::new(p).unwrap();

        let eval_status = Arc::new(Mutex::new(false));
        self.eval_handlers.push(eval_status.clone());
        Dropout {
            bernoulli: bernouli,
            p,
            rng: self.rng.clone(),
            eval: eval_status.clone(),
        }
    }
}