DEV Community

Cover image for Deep Neural Network from Scratch in Rust 🦀- Part 5- Training and Inference

Deep Neural Network from Scratch in Rust 🦀- Part 5- Training and Inference

Akshay Ballal on May 27, 2023

Great!! You have made it to the final part of the series. In this part, we will train our model and test it by building a prediction function. Luck...
Collapse
 
killyosaur profile image
killyosaur

Noticed that none of the branches contains a working version of the neural network and I seem to be missing something as mine isn't getting the accuracy yours is, and I keep getting a NaN out of the cost function :)

Thanks for the article though, I am learning a bunch!

Collapse
 
killyosaur profile image
killyosaur • Edited

found a way to get a working version!

It's still a very low accuracy for the test data though :(

Last Edit: Found the bug.