/CNNforMNIST

A CNN for MNIST Dataset in Python

Primary LanguagePythonMIT LicenseMIT

CNNforMNIST

A CNN for MNIST Dataset, the details could be found at http://www.aaronguan.com/CNNforMNIST.html

Structure

How to read

mainProcess.py: main processing for training and testing.

functionDef.py: simplifying the function: 2-D convolution, maximum pooling, bias and weight.

CNNLayers.py: defines the struture of the CNN: 1 input layer, 2 convolution layers and 2 fully-connected layers.

lossFunction.py: defines the loss function and the accuracy of the CNN.

Tensorboard visualization

main.py embedded the tesorboard visulization function.

log contains the output log of the program

summary contains the tensorboad visulization file

How to visualize?

tensorboard --logdir= "path/to/log-directory"

Results

Accuracy:

Loss: