micrograd-rs 0.0.2

Rust implementation of Andrej Karpathy's micrograd scalar-valued engine with basic neural network.
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

use crate::{layer::Layer, value::Value};

pub struct MLP {
    layers: Vec<Layer>,
}

impl MLP {
    pub fn new(input_count: usize, output_counts: Vec<usize>) -> MLP {
        let output_counts_len = output_counts.len();
        let layer_sizes: Vec<usize> = [input_count].into_iter().chain(output_counts).collect();

        MLP {
            layers: (0..output_counts_len)
                .map(|i| Layer::new(layer_sizes[i], layer_sizes[i + 1]))
                .collect(),
        }
    }

    pub fn forward(&self, mut xs: Vec<Value>) -> Vec<Value> {
        for layer in &self.layers {
            xs = layer.forward(&xs);
        }
        xs
    }

    pub fn parameters(&self) -> Vec<Value> {
        self.layers.iter().flat_map(|l| l.parameters()).collect()
    }
}