diff --git a/README.md b/README.md index 6f15376..de0abc2 100644 --- a/README.md +++ b/README.md @@ -16,9 +16,7 @@ Error rate: 2.95% Error rate: 3.87% # Convolutional Neural Network -Error rate: 3.61% -96.80% -97.87% +Error rate: 0.70% # Compilation ```bash diff --git a/src/dnn.cpp b/src/dnn.cpp index 7ebdd08..da1ee94 100644 --- a/src/dnn.cpp +++ b/src/dnn.cpp @@ -14,7 +14,7 @@ int main() { srand(time(0)); load_dataset(); - NN nn { + NN nn { // Error rate: 3.87% new LayerLinear(28*28, 28), new LayerSigmoid, new LayerLinear(28 , 28), @@ -23,7 +23,7 @@ int main() { new LayerSigmoid, }; - // NN nn { + // NN nn { // Error rate: 0.70% // new LayerConvolutional(1, 20, {24, 24}, {5, 5}), // new LayerSigmoid, // new LayerAveragePooling({12, 12}, {2, 2}),