snail_nn 0.1.0

small neural network libary, running on the cpu with parallelized stochastic gradient descent
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

#![allow(unused)]
use snail_nn::{mat, prelude::*};

fn main() {
    let mut nn = Model::new(&[2, 3, 3, 1]);

    let input = mat!((0, 1), (1, 0), (1, 1), (0, 0));
    let output = mat!((1), (1), (0), (0));

    let batch = TrainingBatch::new(input.clone(), output.clone());

    let mut timer: u128 = 0;
    let mut epoch: u128 = 0;

    loop {
        timer += 1;
        let (w, b) = nn.gradient(&batch);
        nn.learn(w, b, 1.0);
        epoch += 1;

        if timer % 100 == 0 {
            //clear terminal
            print!("{esc}c", esc = 27 as char);
            println!("epoch: {}",epoch);
            println!("cost: {}", nn.cost(&batch));

            input.iter_rows().zip(output.iter_rows()).for_each(|(i, e)| {
                let out = nn.forward(&i);
                print!("in:[{:?}] ", &i);
                print!("expected: {:?} --> ", e);
                print!("out:[{:?}] ", out);
                print!("\n");
            });
        }
    }
}