diff --git a/README.md b/README.md index 5ab4755..e137180 100644 --- a/README.md +++ b/README.md @@ -49,10 +49,31 @@ Non-trainable params: 0 _________________________________________________________________ ``` -An example of the learning curve of this process: - -alt +alt This learning curve is the result of training with a population size of 100, a mating pool size of 15, uniform crossover, and random additive mutation over 200 generations. +We performed the same process with a convolutional neural network, with each individual having the following architecture: + +``` +Model: "sequential_2" +_________________________________________________________________ +Layer (type) Output Shape Param # +================================================================= +conv2d_2 (Conv2D) (None, 28, 28, 8) 40 +_________________________________________________________________ +conv2d_3 (Conv2D) (None, 28, 28, 16) 528 +_________________________________________________________________ +flatten_1 (Flatten) (None, 12544) 0 +_________________________________________________________________ +dense (Dense) (None, 10) 125450 +================================================================= +Total params: 126,018 +Trainable params: 126,018 +Non-trainable params: 0 +_________________________________________________________________ +``` + +alt + ## Reinforcement Learning Pong Player \ No newline at end of file diff --git a/assets/notes/learning-curve-example.png b/assets/notes/ann-learning-curve.png similarity index 100% rename from assets/notes/learning-curve-example.png rename to assets/notes/ann-learning-curve.png diff --git a/mnist-classifier/logs/cnn/cnn-learning-curve.png b/assets/notes/cnn-learning-curve.png similarity index 100% rename from mnist-classifier/logs/cnn/cnn-learning-curve.png rename to assets/notes/cnn-learning-curve.png