Transfer: LunarLander

EvolutionStrategies/mlp_visualizer.py

+import os
+
 from numpy import cos, sin, arctan
 from matplotlib import pyplot
 
@@ -73,6 +75,10 @@ class Layer:
 
 class NeuralNetwork():
     def __init__(self, architecture, weights):
+        if not os.path.isdir('./models'):
+            os.mkdir('./models')
+        if not os.path.isdir('./models/NN_Images'):
+            os.mkdir('./models/NN_Images')
         self.layers = []
         for i in range(len(architecture)):
             if i > 0:

EvolutionStrategies/walker.py

@@ -12,6 +12,10 @@ np.random.seed(42)
 class Walker:
 
     def __init__(self, hidden_layer, bias, version, load_brain, env):
+
+        if not os.path.isdir('./models'):
+            os.mkdir('./models')
+
         self.env = env
         self.version = version
         self.hidden_layer = hidden_layer
@@ -106,8 +110,6 @@ class Walker:
         network.draw(gen)
 
     def save(self):
-        if not os.path.isdir('./models'):
-            os.mkdir('./models')
         with open('./models/model-pedal%d.p' % self.version, 'wb') as fp:
             pickle.dump(self.weights, fp)