Added legend to action mutation plot

EvolutionStrategies/main.py

@@ -9,12 +9,12 @@ BIAS = False
 POP_SIZE = 50
 MUTATION_FACTOR = 0.1  # 0 <= x <= 1
 LEARNING_RATE = 0.03   # 0 <= x <= 1
-GENS = 7000
+GENS = 10000
 MAX_STEPS = 300  # after 1600 steps the Environment gives us a done anyway.
 DECAY_ALPHA = True
 
 VERSION = 100
-TEST_WALKER = False
+TEST_WALKER = True
 LOAD_BRAIN = False
 RENDER_BEST = False
 if TEST_WALKER:

MutateActions/main.py

@@ -10,7 +10,7 @@ MUTATION_FACTOR = 0.2  # 0 <= x <= 1
 GAME_CANCELLED = False
 LOAD_BRAIN = False
 RENDER_BEST = False
-TEST_WALKER = False
+TEST_WALKER = True
 
 if TEST_WALKER:
     LOAD_BRAIN = True
@@ -27,8 +27,9 @@ if __name__ == '__main__':
         plt.title(f'{POP_SIZE}, {MUTATION_FACTOR}')
         plt.xlabel('Episodes')
         plt.ylabel('Rewards')
-        plt.plot(rewards)
-        plt.plot(steps)
+        plt.plot(rewards, label='Rewards')
+        plt.plot(steps, label='Steps')
+        plt.legend(loc='right')
         plt.savefig(f'./models/{POP_SIZE}, {MUTATION_FACTOR}.png')
         plt.show()
         sys.exit(0)