Python get_vars Examples

Example #1

File: helpers.py Project: mverzett/CTagValidation

 def __init__(self, name, xml_filename, **kwargs):
    mva_vars = get_vars(xml_filename)
    variables = [i.name.value() for i in mva_vars]
    self.reader    = ROOT.TMVA.Reader( "!Color:Silent=%s:Verbose=%s" % (kwargs.get('silent','T'), kwargs.get('verbose','F')))
    self.var_map   = {}
    self.name      = name
    self.variables = variables
    self.xml_filename = xml_filename
    for var in variables:
       self.var_map[var] = array.array('f',[0]) 
       self.reader.AddVariable(var, self.var_map[var])
    self.reader.BookMVA(name, xml_filename)

Example #2

File: dump_training_vars_cfg.py Project: Andrej-CMS/cmssw

parser.add_argument('CvsBXML', help='XML Weight file of the Charm vs B training')
args = parser.parse_args()

if not os.path.isfile(args.CvsLXML):
   raise ValueError('File %s does not exists!' % args.CvsLXML)
if not os.path.isfile(args.CvsBXML):
   raise ValueError('File %s does not exists!' % args.CvsBXML)

with open(args.CvsLXML) as CvsLXML:
   sha1_CvsLXML = hashlib.sha1(CvsLXML.read()).hexdigest()

with open(args.CvsBXML) as CvsBXML:
   sha1_CvsBXML = hashlib.sha1(CvsBXML.read()).hexdigest()

CvsLXML_vpset = cms.VPSet(
   *get_vars(args.CvsLXML, False) #False: do not mimick FileInPath functionalities and take the bare path as the file path
    )
CvsBXML_vpset = cms.VPSet(
   *get_vars(args.CvsBXML, False)
    )

with open('%s/src/RecoBTag/CTagging/python/training_settings.py' % os.environ['CMSSW_BASE'], 'w') as output:
   output.write('import FWCore.ParameterSet.Config as cms\n')
   output.write('''
## IMPORTANT!
## This file was automatically generated by RecoBTag/CTagging/test/dump_training_vars_cfg.py
## with input xml files:
##    - C vs L: %s sha1 checksum: %s
##    - C vs B: %s sha1 checksum: %s 

''' % (args.CvsLXML, sha1_CvsLXML, args.CvsBXML, sha1_CvsBXML))

Example #3

File: analyze_varex_trees.py Project: mverzett/CTagValidation

mva = {
   'CvsL' : FunctorFromMVA('c_vs_l', '%s/src/RecoBTag/CTagging/data/c_vs_udsg.weight.xml' % os.environ['CMSSW_BASE']),
   'CvsB' : FunctorFromMVA('c_vs_b', '%s/src/RecoBTag/CTagging/data/c_vs_b.weight.xml' % os.environ['CMSSW_BASE'])
}
njets = {}
def get_info(pset):
   ret = {
      'entry_name' : pset.taggingVarName.value(),
      'default' : pset.default.value(),
      }
   if hasattr(pset, 'idx'):
      ret['idx'] = pset.idx.value()
   return ret

mva_vars_info = dict((i.name.value(), get_info(i)) for i in get_vars('%s/src/RecoBTag/CTagging/data/c_vs_udsg.weight.xml' % os.environ['CMSSW_BASE']))

#import jets for flat trees
jet_maps = {}
with open('flat_jet_map') as infile:
   jet_maps = pickle.load(infile)

infile_names = glob('training_trees/CombinedSV*.root')
vtx_categories = {
   'CombinedSVNoVertexNoSoftLepton' : 2,
   'CombinedSVPseudoVertexNoSoftLepton' : 1,
   'CombinedSVRecoVertexNoSoftLepton' : 0,

   'CombinedSVNoVertexSoftElectron' : 8,
   'CombinedSVPseudoVertexSoftElectron' : 7,
   'CombinedSVRecoVertexSoftElectron' : 6,

Example #4

File: set_plot_ranges.py Project: mverzett/CTagValidation

from RecoBTag.CTagging.helpers import get_vars
import math
import rootpy.io as io
import pickle
import os

mva_vars_cl = set([i.name.value() for i in get_vars('%s/src/RecoBTag/CTagging/data/c_vs_udsg.weight.xml' % os.environ['CMSSW_BASE'])])
mva_vars_cb = set([i.name.value() for i in get_vars('%s/src/RecoBTag/CTagging/data/c_vs_b.weight.xml' % os.environ['CMSSW_BASE'])])
mva_vars_cl.update(mva_vars_cb)
mva_vars_cl.add('vertexCategory')
#automatic set mins and maxes
mms_vars = dict((i, (float('inf'), float('-inf'))) for i in mva_vars_cl)


debug_CvsL_file = io.root_open('trees/ctag_debug_CvsL.root')
debug_CvsL_tree = debug_CvsL_file.tree
for entry in debug_CvsL_tree:
   for name, minmax in mms_vars.iteritems():
      min_val, max_val = minmax
      val = getattr(entry, name)
      min_val = min(min_val, val)
      max_val = max(max_val, val)
      mms_vars[name] = (min_val, max_val)


flat_file = io.root_open('trees/CombinedSV_ALL.root')
flat_tree = flat_file.tree
for entry in debug_CvsL_tree:
   for name, minmax in mms_vars.iteritems():
      min_val, max_val = minmax
      val = getattr(entry, name)

Example #5

File: dump_training_vars_cfg.py Project: p2l1pfp/cmssw

                    help='XML Weight file of the Charm vs B training')
args = parser.parse_args()

if not os.path.isfile(args.CvsLXML):
    raise ValueError('File %s does not exists!' % args.CvsLXML)
if not os.path.isfile(args.CvsBXML):
    raise ValueError('File %s does not exists!' % args.CvsBXML)

with open(args.CvsLXML) as CvsLXML:
    sha1_CvsLXML = hashlib.sha1(CvsLXML.read()).hexdigest()

with open(args.CvsBXML) as CvsBXML:
    sha1_CvsBXML = hashlib.sha1(CvsBXML.read()).hexdigest()

CvsLXML_vpset = cms.VPSet(*get_vars(
    args.CvsLXML, False
)  #False: do not mimick FileInPath functionalities and take the bare path as the file path
                          )
CvsBXML_vpset = cms.VPSet(*get_vars(args.CvsBXML, False))

with open(
        '%s/src/RecoBTag/CTagging/python/training_settings.py' %
        os.environ['CMSSW_BASE'], 'w') as output:
    output.write('import FWCore.ParameterSet.Config as cms\n')
    output.write('''
## IMPORTANT!
## This file was automatically generated by RecoBTag/CTagging/test/dump_training_vars_cfg.py
## with input xml files:
##    - C vs L: %s sha1 checksum: %s
##    - C vs B: %s sha1 checksum: %s