steffen/self-replicating-neural-networks
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

quick fixed

Browse Source
This commit is contained in:
Thomas Gabor 2019-03-02 18:27:33 +01:00
parent 9feb1bd3d9
commit 24728d7910
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
code/methods.py
View File

@ -176,7 +176,7 @@ class Experiment:
@staticmethod @staticmethod
def from_dill(path): def from_dill(path):
with open(path) as dill_file: with open(path, "r") as dill_file:
return dill.load(dill_file) return dill.load(dill_file)
def __init__(self, name=None, id=None): def __init__(self, name=None, id=None):
@ -190,6 +190,7 @@ class Experiment:
self.dir = self.base_dir + "experiments/exp-" + str(self.experiment_name) + "-" + str(self.experiment_id) + "-" + str(self.next_iteration) + "/" self.dir = self.base_dir + "experiments/exp-" + str(self.experiment_name) + "-" + str(self.experiment_id) + "-" + str(self.next_iteration) + "/"
os.mkdir(self.dir) os.mkdir(self.dir)
print("** created " + str(self.dir)) print("** created " + str(self.dir))
return self
def __exit__(self, exc_type, exc_value, traceback): def __exit__(self, exc_type, exc_value, traceback):
self.save(experiment=self) self.save(experiment=self)
@ -211,8 +212,10 @@ if __name__ == '__main__':
network = Network(features, cells, layers, recurrent=use_recurrent) network = Network(features, cells, layers, recurrent=use_recurrent)
r = RecurrentNetwork(network) r = RecurrentNetwork(network)
loss = r.fit(epochs=10) loss = r.fit(epochs=10)
exp.save(rnet=r)
else: else:
network = Network(features, cells, layers, recurrent=use_recurrent) network = Network(features, cells, layers, recurrent=use_recurrent)
ff = FeedForwardNetwork(network) ff = FeedForwardNetwork(network)
loss = ff.fit(epochs=10) loss = ff.fit(epochs=10)
exp.save(ffnet=ff)
print(loss) print(loss)