Skip to content

Commit

Permalink
Added learning curves to README
Browse files Browse the repository at this point in the history
  • Loading branch information
leethologica committed Nov 16, 2020
1 parent 50cbac8 commit 680654a
Show file tree
Hide file tree
Showing 3 changed files with 24 additions and 3 deletions.
27 changes: 24 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,10 +49,31 @@ Non-trainable params: 0
_________________________________________________________________
```

An example of the learning curve of this process:

<img src="./assets/notes/learning-curve-example.png" alt="alt" style="zoom:75%;" />
<img src="./assets/notes/ann-learning-curve.png" alt="alt" style="zoom:75%;" />

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
_________________________________________________________________
```

<img src="./assets/notes/cnn-learning-curve.png" alt="alt" style="zoom:36%;" />

## Reinforcement Learning Pong Player
File renamed without changes
File renamed without changes

0 comments on commit 680654a

Please sign in to comment.