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
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use std::path::Path;

use newron::dataset::Dataset;
use newron::layers::LayerEnum::*;
use newron::loss::{categorical_entropy::CategoricalEntropy};
use newron::metrics::Metric;
use newron::sequential::Sequential;
use newron::optimizers::sgd::SGD;

fn main() {
    // Path to a folder containing the 4 files :
    // 1/ train-images-idx3-ubyte
    // 2/ train-labels-idx1-ubyte
    // 3/ t10k-images-idx3-ubyte
    // 4/ t10k-labels-idx1-ubyte
    let path = Path::new("datasets/fashion_mnist/");

    let dataset = Dataset::from_ubyte(path).unwrap();
    println!("{:?}", dataset);

    let mut model = Sequential::new();
    model.set_seed(99);

    model.add(Dense {
        input_units: dataset.get_number_features(),
        output_units: 256
    });

    model.add(Dropout {prob: 0.2});
    
    model.add(ReLU);

    model.add(Dense {
        input_units: 256,
        output_units: dataset.get_number_targets()
    });

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

    model.summary();

    model.fit(&dataset, 10, true);
}