created separate file for experiment class

This commit is contained in:
Thomas Gabor
2019-03-03 00:35:40 +01:00
parent 1a82a75369
commit 3b0953c741
3 changed files with 90 additions and 99 deletions

View File

@ -7,6 +7,7 @@ import time
import os
import dill
from experiment import Experiment
import itertools
@ -172,38 +173,6 @@ class FeedForwardNetwork(_BaseNetwork):
bar.update()
return losses
class Experiment:
@staticmethod
def from_dill(path):
with open(path, "r") as dill_file:
return dill.load(dill_file)
def __init__(self, name=None, id=None):
self.experiment_id = id or time.time()
this_file = os.path.realpath(__file__)
self.experiment_name = name or os.path.basename(this_file)
self.base_dir = os.path.realpath((os.path.dirname(this_file) + "/..")) + "/"
self.next_iteration = 0
def __enter__(self):
self.dir = self.base_dir + "experiments/exp-" + str(self.experiment_name) + "-" + str(self.experiment_id) + "-" + str(self.next_iteration) + "/"
os.mkdir(self.dir)
print("** created " + str(self.dir))
return self
def __exit__(self, exc_type, exc_value, traceback):
self.save(experiment=self)
self.next_iteration += 1
def save(self, **kwargs):
for name,value in kwargs.items():
with open(self.dir + "/" + str(name) + ".dill", "wb") as dill_file:
dill.dump(value, dill_file)
if __name__ == '__main__':
with Experiment() as exp:
features, cells, layers = 2, 2, 2