Skip to content

phibr0/rusty-micrograd

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

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);

About

like karpathy/micrograd but written in rust

Resources

Stars

Watchers

Forks

Releases

No releases published

Sponsor this project

Packages

No packages published

Languages