Completed
Prediction function
Class Central Classrooms beta
YouTube videos curated by Class Central.
Classroom Contents
Coding a Neural Network from Scratch in Pure JAX - Machine Learning with JAX - Tutorial 3
Automatically move to the next video in the Classroom when playback concludes
- 1 Intro, structuring the code
- 2 MLP initialization function
- 3 Prediction function
- 4 PyTorch MNIST dataset
- 5 PyTorch data loaders
- 6 Training loop
- 7 Adding the accuracy metric
- 8 Visualize the image and prediction
- 9 Small code refactoring
- 10 Visualizing MLP weights
- 11 Visualizing embeddings using t-SNE
- 12 Analyzing dead neurons
- 13 Outro
