newron 0.5.1

A Rust library to train and infer deep learning models.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#[cfg(test)]
mod sequential_tests {
    use newron::sequential::Sequential;
    use newron::layers::LayerEnum::*;
    use newron::loss::{mse::MSE};
    use newron::optimizers::sgd::SGD;
    use newron::metrics::Metric;
    
    #[test]
    fn test_sequential_stacking() {
        let mut model = Sequential::new();

        model.add(Dense {input_units: 20, output_units: 100});
        model.add(ReLU);

        model.compile(MSE{},
            SGD::new(0.002),
            vec![Metric::Accuracy]);

        assert_eq!(model.layers.len(), 2);
    }
}