diff --git a/EvolutionStrategies/LunarLander/100consecutiveEpisodes.png b/EvolutionStrategies/LunarLander/100consecutiveEpisodes.png new file mode 100644 index 0000000000000000000000000000000000000000..57ae2237bda93d55a930416e83f07e21b7ceb3ed Binary files /dev/null and b/EvolutionStrategies/LunarLander/100consecutiveEpisodes.png differ diff --git a/EvolutionStrategies/LunarLander/LunarLander269p.p b/EvolutionStrategies/LunarLander/LunarLander269p.p new file mode 100644 index 0000000000000000000000000000000000000000..8065729d9a576b78d10a1f73c5492e2bc04f8450 Binary files /dev/null and b/EvolutionStrategies/LunarLander/LunarLander269p.p differ diff --git a/EvolutionStrategies/mlp_visualizer.py b/EvolutionStrategies/mlp_visualizer.py index cd10d30f0fec73a25067f71d2923288398dba0b9..8a03b7ceb65cd141e65caf4fe480f8eb277afbac 100644 --- a/EvolutionStrategies/mlp_visualizer.py +++ b/EvolutionStrategies/mlp_visualizer.py @@ -1,3 +1,5 @@ +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: diff --git a/EvolutionStrategies/walker.py b/EvolutionStrategies/walker.py index 2fbd020b3d7ce9ebe3c29a4b8c40f6c6a88245de..40a0f0ceb3b0745f1a8b25c000db040f18daadda 100644 --- a/EvolutionStrategies/walker.py +++ b/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)