ElectronTagger Class Reference

#include <ElectronTagger.h>

Inheritance diagram for ElectronTagger:

Public Member Functions
virtual float	discriminator (const TagInfoHelper &tagInfo) const override
	b-tag a jet based on track-to-jet parameters in the extened info collection More...
	ElectronTagger (const edm::ParameterSet &)
	explicit ctor More...
void	initialize (const JetTagComputerRecord &) override
Public Member Functions inherited from JetTagComputer
const std::vector< std::string > &	getInputLabels () const
	JetTagComputer ()
	JetTagComputer (const edm::ParameterSet &configuration)
float	operator() (const reco::BaseTagInfo &info) const
float	operator() (const TagInfoHelper &helper) const
void	setupDone ()
virtual	~JetTagComputer ()

Private Attributes
const std::string	m_gbrForestLabel
const btag::LeptonSelector	m_selector
const bool	m_useAdaBoost
const bool	m_useCondDB
const bool	m_useGBRForest
const edm::FileInPath	m_weightFile
std::unique_ptr< TMVAEvaluator >	mvaID

Additional Inherited Members
Protected Member Functions inherited from JetTagComputer
virtual float	discriminator (const reco::BaseTagInfo &) const
void	uses (unsigned int id, const std::string &label)
void	uses (const std::string &label)

Detailed Description

Author

P. Demin - UCL, Louvain-la-Neuve - Belgium

Definition at line 16 of file ElectronTagger.h.

Constructor & Destructor Documentation

ElectronTagger::ElectronTagger

(

const edm::ParameterSet &

cfg

)

explicit ctor

Definition at line 14 of file ElectronTagger.cc.

References mvaID, and JetTagComputer::uses().

                                                         :
    m_selector(cfg),
    m_useCondDB(cfg.getParameter<bool>("useCondDB")),
    m_gbrForestLabel(cfg.existsAs<std::string>("gbrForestLabel") ? cfg.getParameter<std::string>("gbrForestLabel") : ""),
    m_weightFile(cfg.existsAs<edm::FileInPath>("weightFile") ? cfg.getParameter<edm::FileInPath>("weightFile") : edm::FileInPath()),
    m_useGBRForest(cfg.existsAs<bool>("useGBRForest") ? cfg.getParameter<bool>("useGBRForest") : false),
    m_useAdaBoost(cfg.existsAs<bool>("useAdaBoost") ? cfg.getParameter<bool>("useAdaBoost") : false)
  {
    uses("seTagInfos");
    mvaID.reset(new TMVAEvaluator());
  }

Member Function Documentation

float ElectronTagger::discriminator ( const TagInfoHelper &  tagInfo ) const

overridevirtual

b-tag a jet based on track-to-jet parameters in the extened info collection

Reimplemented from JetTagComputer.

Definition at line 46 of file ElectronTagger.cc.

References funct::abs(), reco::SoftLeptonProperties::deltaR, reco::SoftLeptonProperties::elec_mva, f, JetTagComputer::TagInfoHelper::get(), i, infinity, info(), SiPixelRawToDigiRegional_cfi::inputs, btag::LeptonSelector::isNegative(), reco::TemplatedSoftLeptonTagInfo< REF >::leptons(), m_selector, mvaID, reco::TemplatedSoftLeptonTagInfo< REF >::properties(), reco::SoftLeptonProperties::ptRel, random, reco::SoftLeptonProperties::ratio, reco::SoftLeptonProperties::sip2dsig, reco::SoftLeptonProperties::sip3dsig, and GlobalPosition_Frontier_DevDB_cff::tag.

                                                                       {
  // default value, used if there are no leptons associated to this jet
  float bestTag = - std::numeric_limits<float>::infinity();
  const reco::CandSoftLeptonTagInfo & info = tagInfo.get<reco::CandSoftLeptonTagInfo>();

  std::mt19937_64 random;
  std::uniform_real_distribution<float> dist(0.f,1.f);

  // if there are multiple leptons, look for the highest tag result
  for (unsigned int i = 0; i < info.leptons(); i++) {
    const reco::SoftLeptonProperties & properties = info.properties(i);
    if (m_selector(properties)) {
    int theSeed=1+round(10000.0*std::abs(properties.deltaR));
    random.seed(theSeed);
    float rndm = dist(random);
    //for negative tagger, flip 50% of the negative signs to positive value
    float sip3dsig = (m_selector.isNegative() && rndm<0.5) ? -properties.sip3dsig : properties.sip3dsig;
    float sip2dsig = (m_selector.isNegative() && rndm<0.5) ? -properties.sip2dsig : properties.sip2dsig;
    
    std::map<std::string,float> inputs;
    inputs["sip3d"] = sip3dsig;
    inputs["sip2d"] = sip2dsig;
    inputs["ptRel"] = properties.ptRel;
    inputs["deltaR"] = properties.deltaR;
    inputs["ratio"] = properties.ratio;
    inputs["mva_e_pi"] = properties.elec_mva;
    
    float tag = mvaID->evaluate(inputs);
    // Transform output between 0 and 1
    tag = (tag+1.0)/2.0;
        if (tag > bestTag)
           bestTag = tag;
    }
  }
  return bestTag;
}

void ElectronTagger::initialize ( const JetTagComputerRecord &  record )

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 26 of file ElectronTagger.cc.

References edm::FileInPath::fullPath(), edm::eventsetup::EventSetupRecord::get(), edm::eventsetup::DependentRecordImplementation< RecordT, ListT >::getRecord(), m_gbrForestLabel, m_useAdaBoost, m_useCondDB, m_useGBRForest, m_weightFile, mvaID, and makeLayoutFileForGui::variables.

{
    // variable names and order need to be the same as in the training
    std::vector<std::string> variables({"sip3d", "sip2d", "ptRel", "deltaR", "ratio", "mva_e_pi"});
    std::vector<std::string> spectators;
    
    if (m_useCondDB)
    {
        const GBRWrapperRcd & gbrWrapperRecord = record.getRecord<GBRWrapperRcd>();
        
        edm::ESHandle<GBRForest> gbrForestHandle;
        gbrWrapperRecord.get(m_gbrForestLabel.c_str(), gbrForestHandle);

        mvaID->initializeGBRForest(gbrForestHandle.product(), variables, spectators, m_useAdaBoost);
    }
    else
        mvaID->initialize("Color:Silent:Error", "BDT", m_weightFile.fullPath(), variables, spectators, m_useGBRForest, m_useAdaBoost);
}

Member Data Documentation

const std::string ElectronTagger::m_gbrForestLabel

private

Definition at line 27 of file ElectronTagger.h.

Referenced by initialize().

const btag::LeptonSelector ElectronTagger::m_selector

private

Definition at line 25 of file ElectronTagger.h.

Referenced by discriminator().

const bool ElectronTagger::m_useAdaBoost

private

Definition at line 30 of file ElectronTagger.h.

Referenced by initialize().

const bool ElectronTagger::m_useCondDB

private

Definition at line 26 of file ElectronTagger.h.

Referenced by initialize().

const bool ElectronTagger::m_useGBRForest

private

Definition at line 29 of file ElectronTagger.h.

Referenced by initialize().

const edm::FileInPath ElectronTagger::m_weightFile

private

Definition at line 28 of file ElectronTagger.h.

Referenced by initialize().

std::unique_ptr<TMVAEvaluator> ElectronTagger::mvaID

private

Definition at line 32 of file ElectronTagger.h.

Referenced by discriminator(), ElectronTagger(), and initialize().