ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/VHbb/python/evaluateMVA.py

Comparing UserCode/VHbb/python/evaluateMVA.py (file contents):
Revision 1.2 by peller, Wed May 9 09:48:23 2012 UTC vs.
Revision 1.18 by nmohr, Wed Nov 28 13:04:59 2012 UTC

#	Line 2 | Line 2
2		import sys
3		import os
4		import ROOT
5	–	from ROOT import TFile
5		from array import array
6		from math import sqrt
7		from copy import copy
8		#suppres the EvalInstace conversion warning bug
9		import warnings
10		warnings.filterwarnings( action='ignore', category=RuntimeWarning, message='creating converter.*' )
11	<	from ConfigParser import SafeConfigParser
12	<	from samplesinfo import sample
14	<	from mvainfos import mvainfo
11	>	from optparse import OptionParser
12	>	from BetterConfigParser import BetterConfigParser
13		import pickle
16	–	from progbar import progbar
17	–	from printcolor import printc
14
15		#CONFIGURE
16	<
16	>	ROOT.gROOT.SetBatch(True)
17	>	print 'hello'
18		#load config
19	<	config = SafeConfigParser()
20	<	config.read('./config')
19	>	#os.mkdir(path+'/sys')
20	>	argv = sys.argv
21	>	parser = OptionParser()
22	>	parser.add_option("-U", "--update", dest="update", default=0,
23	>	help="update infofile")
24	>	parser.add_option("-D", "--discr", dest="discr", default="",
25	>	help="discriminators to be added")
26	>	#parser.add_option("-I", "--inpath", dest="inpath", default="",
27	>	#                      help="path to samples")
28	>	#parser.add_option("-O", "--outpath", dest="outpath", default="",
29	>	#                      help="path where to store output samples")
30	>	parser.add_option("-S", "--samples", dest="names", default="",
31	>	help="samples you want to run on")
32	>	parser.add_option("-C", "--config", dest="config", default=[], action="append",
33	>	help="configuration file")
34	>	(opts, args) = parser.parse_args(argv)
35	>
36	>	from samplesclass import sample
37	>	from mvainfos import mvainfo
38	>	from progbar import progbar
39	>	from printcolor import printc
40	>
41	>	if opts.config =="":
42	>	opts.config = "config"
43	>	config = BetterConfigParser()
44	>	#config.read('./config7TeV_ZZ')
45	>	config.read(opts.config)
46	>	anaTag = config.get("Analysis","tag")
47
48		#get locations:
49		Wdir=config.get('Directories','Wdir')
50	<
50	>	MVASubdir=config.get('Directories','MVAdir')
51	>	samplesinfo=config.get('Directories','samplesinfo')
52
53		#systematics
30	–	systematics=config.get('systematics','systematics')
31	–	systematics=systematics.split(' ')
54
55	<	#TreeVar Array
56	<	MVA_Vars={}
57	<	for systematic in systematics:
58	<	MVA_Vars[systematic]=config.get('treeVars',systematic)
59	<	MVA_Vars[systematic]=MVA_Vars[systematic].split(' ')
60	<
61	<	######################
62	<	#Evaluate multi: Must Have same treeVars!!!
63	<
64	<	Apath=sys.argv[1]
65	<	arglist=sys.argv[2] #RTight_blavla,bsbsb
66	<
67	<	#for axample
46	<	#0 5 0
47	<	#and
48	<	#5 -1 1
49	<
50	<	start=int(sys.argv[3])
51	<	stop=int(sys.argv[4])
52	<	doinfo=bool(int(sys.argv[5]))
55	>
56	>	INpath = config.get('Directories','MVAin')
57	>	OUTpath = config.get('Directories','MVAout')
58	>
59	>	infofile = open(samplesinfo,'r')
60	>	info = pickle.load(infofile)
61	>	infofile.close()
62	>	arglist=opts.discr #RTight_blavla,bsbsb
63	>
64	>	namelistIN=opts.names
65	>	namelist=namelistIN.split(',')
66	>
67	>	doinfo=bool(int(opts.update))
68
69		MVAlist=arglist.split(',')
70
71	+
72		#CONFIG
73		#factory
74		factoryname=config.get('factory','factoryname')
75		#MVA
60	–	#MVAnames=[]
61	–	#for MVA in MVAlist:
62	–	#    print MVA
63	–	#    MVAnames.append(config.get(MVA,'MVAname'))
64	–	#print Wdir+'/weights/'+factoryname+'_'+MVAname+'.info'
65	–	#MVAinfofiles=[]
76		MVAinfos=[]
77	+	MVAdir=config.get('Directories','vhbbpath')
78		for MVAname in MVAlist:
79	<	MVAinfofile = open(Wdir+'/weights/'+factoryname+'_'+MVAname+'.info','r')
79	>	MVAinfofile = open(MVAdir+'/data/'+factoryname+'_'+MVAname+'.info','r')
80		MVAinfos.append(pickle.load(MVAinfofile))
81		MVAinfofile.close()
82
72	–	treeVarSet=MVAinfos[0].varset
73	–	#variables
74	–	#TreeVar Array
75	–	MVA_Vars={}
76	–	for systematic in systematics:
77	–	MVA_Vars[systematic]=config.get(treeVarSet,systematic)
78	–	MVA_Vars[systematic]=MVA_Vars[systematic].split(' ')
79	–	#Spectators:
80	–	#spectators=config.get(treeVarSet,'spectators')
81	–	#spectators=spectators.split(' ')
82	–	#progbar quatsch
83		longe=40
84		#Workdir
85		workdir=ROOT.gDirectory.GetPath()
86	–	#os.mkdir(Apath+'/MVAout')
87	–
88	–	#Book TMVA readers: MVAlist=["MMCC_bla","CC5050_bla"]
89	–	readers=[]
90	–	for MVA in MVAlist:
91	–	readers.append(ROOT.TMVA.Reader("!Color:!Silent"))
92	–
93	–	#define variables and specatators
94	–	MVA_var_buffer = []
95	–	for i in range(len( MVA_Vars['Nominal'])):
96	–	MVA_var_buffer.append(array( 'f', [ 0 ] ))
97	–	for reader in readers:
98	–	reader.AddVariable( MVA_Vars['Nominal'][i],MVA_var_buffer[i])
99	–	#MVA_spectator_buffer = []
100	–	#for i in range(len(spectators)):
101	–	#    MVA_spectator_buffer.append(array( 'f', [ 0 ] ))
102	–	#    for reader in readers:
103	–	#        reader.AddSpectator(spectators[i],MVA_spectator_buffer[i])
104	–	#Load raeder
105	–	for i in range(0,len(readers)):
106	–	readers[i].BookMVA(MVAinfos[i].MVAname,MVAinfos[i].getweightfile())
107	–	#--> Now the MVA is booked
86
87		#Apply samples
88	<	infofile = open(Apath+'/samples.info','r')
88	>	infofile = open(samplesinfo,'r')
89		Ainfo = pickle.load(infofile)
90		infofile.close()
91
114	–	#eval
115	–	for job in Ainfo[start:stop]:
116	–	#get trees:
117	–	input = TFile.Open(job.getpath(),'read')
118	–	outfile = TFile.Open(job.path+'/MVAout2/'+job.prefix+job.identifier+'.root','recreate')
119	–	input.cd()
120	–	obj = ROOT.TObject
121	–	for key in ROOT.gDirectory.GetListOfKeys():
122	–	input.cd()
123	–	obj = key.ReadObj()
124	–	print obj.GetName()
125	–	if obj.GetName() == job.tree:
126	–	continue
127	–	outfile.cd()
128	–	print key.GetName()
129	–	obj.Write(key.GetName())
130	–	tree = input.Get(job.tree)
131	–	nEntries = tree.GetEntries()
132	–	outfile.cd()
133	–	newtree = tree.CloneTree(0)
134	–
135	–	#MCs:
136	–	if job.type != 'DATA':
137	–	MVA_formulas={}
138	–	for systematic in systematics:
139	–	#print '\t\t - ' + systematic
140	–	MVA_formulas[systematic]=[]
141	–	#create TTreeFormulas
142	–	for j in range(len( MVA_Vars['Nominal'])):
143	–	MVA_formulas[systematic].append(ROOT.TTreeFormula("MVA_formula%s_%s"%(j,systematic),MVA_Vars[systematic][j],tree))
144	–	outfile.cd()
145	–	#Setup Branches
146	–	MVAbranches=[]
147	–	for i in range(0,len(readers)):
148	–	MVAbranches.append(array('f',[0]*9))
149	–	newtree.Branch(MVAinfos[i].MVAname,MVAbranches[i],'nominal:JER_up:JER_down:JES_up:JES_down:beff_up:beff_down:bmis_up:bmis_down/F')
150	–	print '\n--> ' + job.name +':'
151	–	#progbar setup
152	–	if nEntries >= longe:
153	–	step=int(nEntries/longe)
154	–	long=longe
155	–	else:
156	–	long=nEntries
157	–	step = 1
158	–	bar=progbar(long)
159	–	#Fill event by event:
160	–	for entry in range(0,nEntries):
161	–	if entry % step == 0:
162	–	bar.move()
163	–	#load entry
164	–	tree.GetEntry(entry)
165	–	for systematic in systematics:
166	–	for j in range(len( MVA_Vars['Nominal'])):
167	–	MVA_var_buffer[j][0] = MVA_formulas[systematic][j].EvalInstance()
168	–
169	–	for j in range(0,len(readers)):
170	–	MVAbranches[j][systematics.index(systematic)] = readers[j].EvaluateMVA(MVAinfos[j].MVAname)
171	–	#Fill:
172	–	newtree.Fill()
173	–	newtree.AutoSave()
174	–	outfile.Close()
175	–
176	–	#DATA:
177	–	if job.type == 'DATA':
178	–	#MVA Formulas
179	–	MVA_formulas_Nominal = []
180	–	#create TTreeFormulas
181	–	for j in range(len( MVA_Vars['Nominal'])):
182	–	MVA_formulas_Nominal.append(ROOT.TTreeFormula("MVA_formula%s_Nominal"%j, MVA_Vars['Nominal'][j],tree))
183	–	outfile.cd()
184	–	MVAbranches=[]
185	–	for i in range(0,len(readers)):
186	–	MVAbranches.append(array('f',[0]))
187	–	newtree.Branch(MVAinfos[i].MVAname,MVAbranches[i],'nominal/F')
188	–	#progbar
189	–	print '\n--> ' + job.name +':'
190	–	if nEntries >= longe:
191	–	step=int(nEntries/longe)
192	–	long=longe
193	–	else:
194	–	long=nEntries
195	–	step = 1
196	–	bar=progbar(long)
197	–	#Fill event by event:
198	–	for entry in range(0,nEntries):
199	–	if entry % step == 0:
200	–	bar.move()
201	–	#load entry
202	–	tree.GetEntry(entry)
203	–	#nominal:
204	–	for j in range(len( MVA_Vars['Nominal'])):
205	–	MVA_var_buffer[j][0] = MVA_formulas_Nominal[j].EvalInstance()
206	–
207	–	for j in range(0,len(readers)):
208	–	MVAbranches[j][0]= readers[j].EvaluateMVA(MVAinfos[j].MVAname)
209	–	newtree.Fill()
210	–	newtree.AutoSave()
211	–	outfile.Close()
212	–
92
93	+	class MvaEvaluater:
94	+	def __init__(self, config, MVAinfo):
95	+	self.varset = MVAinfo.varset
96	+	#Define reader
97	+	self.reader = ROOT.TMVA.Reader("!Color:!Silent")
98	+	MVAdir=config.get('Directories','vhbbpath')
99	+	self.systematics=config.get('systematics','systematics').split(' ')
100	+	self.MVA_Vars={}
101	+	self.MVAname = MVAinfo.MVAname
102	+	for systematic in self.systematics:
103	+	self.MVA_Vars[systematic]=config.get(self.varset,systematic)
104	+	self.MVA_Vars[systematic]=self.MVA_Vars[systematic].split(' ')
105	+	#define variables and specatators
106	+	self.MVA_var_buffer = []
107	+	for i in range(len( self.MVA_Vars['Nominal'])):
108	+	self.MVA_var_buffer.append(array( 'f', [ 0 ] ))
109	+	self.reader.AddVariable( self.MVA_Vars['Nominal'][i],self.MVA_var_buffer[i])
110	+	self.reader.BookMVA(MVAinfo.MVAname,MVAdir+'/data/'+MVAinfo.getweightfile())
111	+	#--> Now the MVA is booked
112	+
113	+	def setBranches(self,tree,job):
114	+	#Set formulas for all vars
115	+	self.MVA_formulas={}
116	+	for systematic in self.systematics:
117	+	if job.type == 'DATA' and not systematic == 'Nominal': continue
118	+	self.MVA_formulas[systematic]=[]
119	+	for j in range(len( self.MVA_Vars['Nominal'])):
120	+	self.MVA_formulas[systematic].append(ROOT.TTreeFormula("MVA_formula%s_%s"%(j,systematic),self.MVA_Vars[systematic][j],tree))
121	+
122	+	def evaluate(self,MVAbranches,job):
123	+	#Evaluate all vars and fill the branches
124	+	for systematic in self.systematics:
125	+	for j in range(len( self.MVA_Vars['Nominal'])):
126	+	if job.type == 'DATA' and not systematic == 'Nominal': continue
127	+	self.MVA_var_buffer[j][0] = self.MVA_formulas[systematic][j].EvalInstance()
128	+	MVAbranches[self.systematics.index(systematic)] = self.reader.EvaluateMVA(self.MVAname)
129	+
130	+
131	+	theMVAs = []
132	+	for mva in MVAinfos:
133	+	theMVAs.append(MvaEvaluater(config,mva))
134
135
136	+	#eval
137	+	for job in Ainfo:
138	+	if eval(job.active):
139	+	if job.name in namelist:
140	+	#get trees:
141	+	print INpath+'/'+job.prefix+job.identifier+'.root'
142	+	input = ROOT.TFile.Open(INpath+'/'+job.prefix+job.identifier+'.root','read')
143	+	print OUTpath+'/'+job.prefix+job.identifier+'.root'
144	+	outfile = ROOT.TFile.Open(OUTpath+'/'+job.prefix+job.identifier+'.root','recreate')
145	+	input.cd()
146	+	obj = ROOT.TObject
147	+	for key in ROOT.gDirectory.GetListOfKeys():
148	+	input.cd()
149	+	obj = key.ReadObj()
150	+	#print obj.GetName()
151	+	if obj.GetName() == job.tree:
152	+	continue
153	+	outfile.cd()
154	+	#print key.GetName()
155	+	obj.Write(key.GetName())
156	+	tree = input.Get(job.tree)
157	+	nEntries = tree.GetEntries()
158	+	outfile.cd()
159	+	newtree = tree.CloneTree(0)
160	+
161	+
162	+	#Set branch adress for all vars
163	+	for i in range(0,len(theMVAs)):
164	+	theMVAs[i].setBranches(tree,job)
165	+	outfile.cd()
166	+	#Setup Branches
167	+	MVAbranches=[]
168	+	for i in range(0,len(theMVAs)):
169	+	if job.type == 'Data':
170	+	MVAbranches.append(array('f',[0]))
171	+	newtree.Branch(MVAinfos[i].MVAname,MVAbranches[i],'nominal/F')
172	+	else:
173	+	MVAbranches.append(array('f',[0]*11))
174	+	newtree.Branch(theMVAs[i].MVAname,MVAbranches[i],'nominal:JER_up:JER_down:JES_up:JES_down:beff_up:beff_down:bmis_up:bmis_down:beff1_up:beff1_down/F')
175	+	MVA_formulas_Nominal = []
176	+	print '\n--> ' + job.name +':'
177	+	#progbar setup
178	+	if nEntries >= longe:
179	+	step=int(nEntries/longe)
180	+	long=longe
181	+	else:
182	+	long=nEntries
183	+	step = 1
184	+	bar=progbar(long)
185	+	#Fill event by event:
186	+	for entry in range(0,nEntries):
187	+	if entry % step == 0:
188	+	bar.move()
189	+	#load entry
190	+	tree.GetEntry(entry)
191	+
192	+	for i in range(0,len(theMVAs)):
193	+	theMVAs[i].evaluate(MVAbranches[i],job)
194	+	#Fill:
195	+	newtree.Fill()
196	+	newtree.AutoSave()
197	+	outfile.Close()
198	+
199	+	print '\n'
200
201		#Update Info:
202		if doinfo:
203		for job in Ainfo:
204		for MVAinfo in MVAinfos:
205		job.addcomment('Added MVA %s'%MVAinfo.MVAname)
206	<	job.addpath('/MVAout')
207	<	infofile = open(Apath+'/MVAout/samples.info','w')
206	>	job.addpath(MVAdir)
207	>	infofile = open(samplesinfo,'w')
208		pickle.dump(Ainfo,infofile)
209		infofile.close()
210

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines