init

1 files changed, 89 insertions, 0 deletions

diff --git a/run.py b/run.py
new file mode 100755
index 0000000..cf9235c
--- /dev/null
+++ b/run.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python2
+
+import os
+import datetime
+import logging
+from src.evaluator import Evaluator
+from src.smt_semparse_config import SMTSemparseConfig
+from src.smt_semparse_experiment import SMTSemparseExperiment
+
+LOGFILE_NAME = 'run.log'
+
+def run_one(config):
+  # create work dir for this run
+  # moses can't handle paths with colons
+  timestamp = datetime.datetime.now().strftime('%Y-%m-%dT%H.%M.%S')
+  run_work_dir = os.path.join(base_work_dir, timestamp)
+  assert not os.path.exists(run_work_dir)
+  os.makedirs(run_work_dir)
+  config.put('work_dir', run_work_dir)
+  if os.path.exists('latest'):
+    os.remove('latest')
+  os.symlink(run_work_dir, 'latest')
+
+  # set up logging
+  if config.run == 'debug':
+    logging.basicConfig(level=logging.DEBUG)
+  else:
+    log_path = os.path.join(run_work_dir, LOGFILE_NAME)
+    logging.basicConfig(filename=log_path, level=logging.INFO)
+
+  experiment = SMTSemparseExperiment(config)
+  if config.run == 'debug':
+    experiment.run_fold(1)
+  elif config.run == 'dev':
+    for i in range(10):
+      experiment.run_fold(i)
+  elif config.run == 'test':
+    experiment.run_split()
+  else:
+    assert False
+
+  if not config.nlg:
+    logging.info('evaluating')
+    Evaluator(config).run()
+
+if __name__ == '__main__':
+
+  # load config
+  config = SMTSemparseConfig('settings.yaml', 'dependencies.yaml')
+
+  # create base work dir if it doesn't exist
+  base_work_dir = os.path.join(config.smt_semparse, config.workdir)
+  if not os.path.exists(base_work_dir):
+    os.makedirs(base_work_dir)
+
+  run_one(config)
+
+  #for np in (True, False):
+  #  for model in ('phrase', 'hier'):
+  #    for lang in ('en', 'de', 'el', 'th'):
+  #      print 'np: %s\nmodel: %s\nlang: %s' % (np, model, lang)
+  #      config.put('np', np)
+  #      config.put('model', model)
+  #      config.put('lang', lang)
+  #      if lang == 'en':
+  #        config.put('symm', 'srctotgt')
+  #      else:
+  #        config.put('symm', 'tgttosrc')
+  #      for i in range(10):
+  #        run_one(config)
+
+  #for lfrac in (0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0):
+  ##for lfrac in (0.1,):
+  #  for monolingual in (True, False):
+  #  #for monolingual in (True,):
+  #    for ul_only in (True, False):
+  #    #for ul_only in (True,):
+  #      if (not monolingual) and ul_only:
+  #        continue
+  #      config.put('lfrac', lfrac)
+  #      config.put('monolingual', monolingual)
+  #      config.put('ul_only', ul_only)
+  #      print 'lfrac: %f' % lfrac
+  #      print 'monolingual: %s' % monolingual
+  #      print 'ul_only: %s' % ul_only
+  #      for i in range(10):
+  #      #for i in range(2):
+  #        run_one(config)
+