/NN-with-cifar10-test-2

Another trial using different activation function

Primary LanguageJupyter Notebook

NN_with_cifar10_test_2

Another trial using different activation function

Setting:

Structure of NN: Fully connected

Activation function: Linear, then ReLU

Number of hidden layers: 4

Nodes within each layer: 1024

Loss function: Cross-Entropy

Software used for training: Keras

Data set: Cifar-10

Results:

Number of observations used for training: 128 (minibatch size)

Training time: 1 min, 18 s

Test accuracy: 0.44