-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodelNames.py
70 lines (70 loc) · 2.24 KB
/
modelNames.py
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
def getModelNames(category="blah"):
return [
"bkg_dijetsimple2",
"bkg_dijet2",
"bkg_dijet3",
"bkg_atlas1",
"bkg_atlas2",
"bkg_atlas3",
"bkg_expow1",
"bkg_expow2",
"bkg_expow3",
"bkg_cheb1",
"bkg_cheb2",
"bkg_cheb3",
"bkg_exp1",
"bkg_exp3",
"bkg_pow1",
"bkg_pow3",
"bkg_lau1",
"bkg_lau2",
"bkg_lau3",
"bkg_vvdijet1",
"bkg_vvdijet2"
]
def getGoodModelNames(category):
if not category in ["btag", "antibtag"]:
print "error: got a bad category in getGoodModelNames: %s" % category
exit(1)
elif category == "btag":
return [
"bkg_dijetsimple2", # "bkg_dijetsimple2",
#"bkg_dijet2", # #"bkg_dijet2",
#"bkg_dijet3", # #"bkg_dijet3",
"bkg_atlas1", # "bkg_atlas1",
#"bkg_atlas2", # "bkg_atlas2",
#"bkg_atlas3", # "bkg_atlas3",
"bkg_expow1", # "bkg_expow1",
"bkg_expow2", # "bkg_expow2",
#"bkg_expow3", # "bkg_expow3",
#"bkg_pow1", # ##"bkg_pow1",
#"bkg_pow3", # #"bkg_pow3",
"bkg_exp1", # "bkg_exp1",
#"bkg_exp3", # #"bkg_exp3",
#"bkg_lau1", # ##"bkg_lau1",
#"bkg_lau2", # ##"bkg_lau2",
#"bkg_lau3", # ##"bkg_lau3",
"bkg_vvdijet1", # #"bkg_vvdijet1",
"bkg_vvdijet2" # "bkg_vvdijet2"
]
elif category == "antibtag":
return [
"bkg_dijetsimple2", # "bkg_dijetsimple2",
#"bkg_dijet2", # #"bkg_dijet2",
"bkg_dijet3", # #"bkg_dijet3",
"bkg_atlas1", # "bkg_atlas1",
#"bkg_atlas2", # "bkg_atlas2",
#"bkg_atlas3", # "bkg_atlas3",
"bkg_expow1", # "bkg_expow1",
"bkg_expow2", # "bkg_expow2",
#"bkg_expow3", # "bkg_expow3",
#"bkg_pow1", # #"bkg_pow1",
#"bkg_pow3", # #"bkg_pow3",
"bkg_exp1", # "bkg_exp1",
#"bkg_exp3", # #"bkg_exp3",
#"bkg_lau1", # #"bkg_lau1",
#"bkg_lau2", # ##"bkg_lau2",
#"bkg_lau3", # ##"bkg_lau3",
"bkg_vvdijet1", # "bkg_vvdijet1"
"bkg_vvdijet2" # #"bkg_vvdijet2"
]