Skip to content

aaronzguan/CNNforMNIST

Repository files navigation

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: