Wed 20 Mar 12:11:23 CDT 2019

1 files changed, 77 insertions, 0 deletions

diff --git a/submitter/contour_emcee_dag.py b/submitter/contour_emcee_dag.py
new file mode 100644
index 0000000..b16312a
--- /dev/null
+++ b/submitter/contour_emcee_dag.py
@@ -0,0 +1,77 @@
+#! /usr/bin/env python
+
+import os
+import numpy as np
+
+gfsource      = os.environ['GOLEMSOURCEPATH'] + '/GolemFit'
+condor_script = gfsource + '/scripts/flavour_ratio/submitter/contour_emcee_submit.sub'
+
+injected_ratios = [
+    (1, 1, 1),
+    (1, 0, 0),
+    (0, 1, 0),
+    (0, 0, 1)
+]
+
+GLOBAL_PARAMS = {}
+
+GLOBAL_PARAMS.update(dict(
+    threads = 1,
+))
+
+# Emcee
+GLOBAL_PARAMS.update(dict(
+    run_mcmc = 'True',
+    burnin   = 250,
+    nsteps   = 500,
+    nwalkers = 60,
+    seed     = 25,
+    mcmc_seed_type = 'uniform'
+))
+
+# Likelihood
+GLOBAL_PARAMS.update(dict(
+    likelihood  = 'golemfit',
+))
+
+# GolemFit
+GLOBAL_PARAMS.update(dict(
+    ast  = 'p2_0',
+    # data = 'realisation'
+    # data = 'asimov'
+    data = 'real'
+))
+
+# Plot
+GLOBAL_PARAMS.update(dict(
+    plot_angles   = 'False',
+    plot_elements = 'False',
+))
+
+outfile = 'dagman_FR_CONTOUR_EMCEE_{0}'.format(GLOBAL_PARAMS['data'])
+outfile += 'more_sys_flat'
+outfile += '.submit'
+
+output  = '/data/user/smandalia/flavour_ratio/data/contour_emcee/{0}/{1}/'.format(
+    GLOBAL_PARAMS['likelihood'], GLOBAL_PARAMS['data']
+)
+# output += 'more_sys/'
+output += 'more_sys_flat/'
+# output += 'noprompt/'
+# output += 'strictpriors/'
+
+with open(outfile, 'w') as f:
+    job_number = 1
+    for inj in injected_ratios:
+        print 'inj', inj
+        f.write('JOB\tjob{0}\t{1}\n'.format(job_number, condor_script))
+        f.write('VARS\tjob{0}\tir0="{1}"\n'.format(job_number, inj[0]))
+        f.write('VARS\tjob{0}\tir1="{1}"\n'.format(job_number, inj[1]))
+        f.write('VARS\tjob{0}\tir2="{1}"\n'.format(job_number, inj[2]))
+        for key in GLOBAL_PARAMS.iterkeys():
+            f.write('VARS\tjob{0}\t{1}="{2}"\n'.format(job_number, key, GLOBAL_PARAMS[key]))
+        f.write('VARS\tjob{0}\toutfile="{1}"\n'.format(job_number, output))
+        job_number += 1
+        if GLOBAL_PARAMS['data'] == 'real': break
+
+print 'dag file = {0}'.format(outfile)