newron 0.4.0

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

#[cfg(test)]
mod sequential_tests {
    use newron::sequential::Sequential;
    use newron::layers::{dense::Dense, relu::ReLU};
    
    #[test]
    fn test_sequential_stacking() {
        let mut model = Sequential::new();

        model.add(Dense::new(20, 30));
        model.add(ReLU);

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