An open API service indexing awesome lists of open source software.

https://github.com/phibr0/rusty-micrograd

like karpathy/micrograd but written in rust
https://github.com/phibr0/rusty-micrograd

Last synced: 9 months ago
JSON representation

like karpathy/micrograd but written in rust

Awesome Lists containing this project

README

          

# rusty micrograd

Predictions made by a trained model on the mnist dataset

I built this after my "Probabilistic Reasoning and Machine Learning" course.
Learns handwritten digit (mnist) classification from scratch.

```rust
let mut model = Model::new(
&[784, 32, 16, 10],
&[Activation::ReLU, Activation::ReLU, Activation::Softmax],
);

let epochs = 100;
let eta = 0.2;
model.train(&train_data, epochs, eta, Loss::CrossEntropy);
```