Zero-dependency neural network trained to recognize handwritten digits from the MNIST dataset. The stored src/mnist-model.json file has an accuracy of 94.8% on the test set. The training performance is not very good. :)
npm i
Download the MNIST dataset from here and extract it like so:
src/dataset/mnist_train.csv
src/dataset/mnist_test.csv
npm run mnist:train <epochs> <learning_rate> <batch_size>
USE_EXISTING=true npm run mnist:train <epochs> <learning_rate> <batch_size>
npm run mnist:test