Add md5 for hashing config file

remove date from work_dir name

Author: DesmondYuan

pertbio/pertbio/__init__.py

@@ -4,4 +4,5 @@
 from pertbio.model import *
 from pertbio.dataset import *
 from pertbio.train import *
+from pertbio.utils import *
 from pertbio.version import __version__, VERSION

pertbio/pertbio/utils.py

@@ -1,6 +1,7 @@
 import tensorflow as tf
 import numpy as np
 import time
+import hashlib
 
 def loss(x_gold, x_hat,l1, W):
     with tf.variable_scope("loss", reuse=True):
@@ -37,3 +38,9 @@ def log(self, s):
             print("#" * 4 * self.hierachy, " ", s, "  --time elapsed: %.2f"%(time.time() - self.time))
             self.time = time.time()
             self.step_count += 1
+
+def md5(key):
+    '''
+    returns a hashed with md5 string of the key
+    '''
+    return hashlib.md5(key.encode()).hexdigest()

scripts/main.py

@@ -29,8 +29,8 @@ def prepare_workdir(cfg):
 
     # Create Output Folder
     try:
-        date = datetime.date.today().strftime("%y%m%d")
-        experiment_path = 'results/{}_{}'.format(cfg.experiment_id, date)
+        md5 = pertbio.utils.md5(str(vars(cfg)))
+        experiment_path = 'results/{}_{}'.format(cfg.experiment_id, md5)
         os.makedirs(experiment_path)
     except:
         pass
@@ -72,9 +72,9 @@ def prepare_workdir(cfg):
     except:
         seed = int(working_index) + 1000
         set_seed(seed)
-    
+
     cfg.dataset = prepare_workdir(cfg)
-  
+
     logger = pertbio.utils.time_logger(time_logger_step = 1, hierachy = 3)
     args = cfg
     for i, stage in enumerate(cfg.stages):