1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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)
|
