gotch/example/mnist
2020-06-23 15:21:16 +10:00
..
cnn.go WIP(example/mnist): conv 2020-06-23 01:07:07 +10:00
linear.go fix(nn/sequential, linear): fixed memory held up due to undeleted middle tensors 2020-06-23 13:37:26 +10:00
main.go WIP(example/mnist): nn 2020-06-18 17:14:48 +10:00
nn.go fix(nn/sequential): fixed wrong index at Forward method 2020-06-23 15:21:16 +10:00
README.md WIP(example/mnist): nn 2020-06-18 17:14:48 +10:00

Linear Regression, NN, and CNN on MNIST dataset

MNIST

  • MNIST files can be obtained from this source and put in data/mnist from root folder of this project.

  • Load MNIST data using helper function at vision sub-package

Linear Regression

  • Run with go clean -cache -testcache && go run . -model="linear"

  • Accuraccy should be about 91.68%.

Neural Network (NN)

  • Run with go clean -cache -testcache && go run . -model="nn"

  • Accuraccy should be about TODO: update%.

Convolutional Neural Network (CNN)

  • Run with go clean -cache -testcache && go run . -model="cnn"

  • Accuraccy should be about TODO: update%.