#include <ElectronMVAEstimatorRun2.h>

Inheritance diagram for ElectronMVAEstimatorRun2:

Public Member Functions
	ElectronMVAEstimatorRun2 (const edm::ParameterSet &conf)

	ElectronMVAEstimatorRun2 (const std::string &mvaTag, const std::string &mvaName, int nCategories, const std::string &variableDefinition, const std::vector< std::string > &categoryCutStrings, const std::vector< std::string > &weightFileNames, bool debug=false)

int	findCategory (const reco::Candidate *candidate) const override

float	mvaValue (const reco::Candidate *candidate, std::vector< float > const &auxVariables, int &iCategory) const override

	~ElectronMVAEstimatorRun2 () override

Public Member Functions inherited from AnyMVAEstimatorRun2Base
	AnyMVAEstimatorRun2Base (const edm::ParameterSet &conf)

	AnyMVAEstimatorRun2Base (const ::std::string &mvaName, const ::std::string &mvaTag, int nCategories, bool debug)

const std::string &	getName () const

int	getNCategories () const

const std::string &	getTag () const

bool	isDebug () const

float	mvaValue (const reco::Candidate *candidate, std::vector< float > const &auxVariables) const

virtual	~AnyMVAEstimatorRun2Base ()

Static Public Member Functions
static std::vector< float >	getExtraVars (reco::GsfElectron const &ele, reco::ConversionCollection const conversions, reco::BeamSpot const beamSpot, double rho)

Private Member Functions
int	findCategory (reco::GsfElectron const &electron) const

void	init (const std::vector< std::string > &weightFileNames)

Private Attributes
std::vector< ThreadSafeStringCut< StringCutObjectSelector< reco::GsfElectron >, reco::GsfElectron > >	categoryFunctions_

std::vector< std::unique_ptr< const GBRForest > >	gbrForests_

MVAVariableManager< reco::GsfElectron >	mvaVarMngr_

std::vector< int >	nVariables_

std::vector< std::vector< int > >	variables_

Detailed Description

Definition at line 16 of file ElectronMVAEstimatorRun2.h.

Constructor & Destructor Documentation

ElectronMVAEstimatorRun2::ElectronMVAEstimatorRun2 ( const edm::ParameterSet & conf )

Definition at line 3 of file ElectronMVAEstimatorRun2.cc.

References categoryFunctions_, AnyMVAEstimatorRun2Base::getNCategories(), edm::ParameterSet::getParameter(), AnyMVAEstimatorRun2Base::getTag(), mps_fire::i, init(), and mvaElectronID_Fall17_iso_V1_cff::weightFileNames.

Referenced by ~ElectronMVAEstimatorRun2().

                                                                              :
   AnyMVAEstimatorRun2Base(conf),
   mvaVarMngr_(conf.getParameter<std::string>("variableDefinition"))
 {
 
   const auto weightFileNames = conf.getParameter<std::vector<std::string> >("weightFileNames");
   const auto categoryCutStrings = conf.getParameter<std::vector<std::string> >("categoryCuts");
 
   if( (int)(categoryCutStrings.size()) != getNCategories() )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of category cuts in ElectronMVAEstimatorRun2" << getTag() << std::endl;
 
   for (int i = 0; i < getNCategories(); ++i) {
       categoryFunctions_.emplace_back(categoryCutStrings[i]);
   }
 
   // Initialize GBRForests from weight files
   init(weightFileNames);
 }

ElectronMVAEstimatorRun2::~ElectronMVAEstimatorRun2 ( )

inlineoverride

Definition at line 22 of file ElectronMVAEstimatorRun2.h.

References debug, ElectronMVAEstimatorRun2(), taus_updatedMVAIds_cff::mvaName, mvaElectronID_Fall17_iso_V1_cff::mvaTag, mvaElectronID_Fall17_iso_V1_cff::nCategories, AlCaHLTBitMon_QueryRunRegistry::string, mvaElectronID_Fall17_iso_V1_cff::variableDefinition, and mvaElectronID_Fall17_iso_V1_cff::weightFileNames.

22 {};

ElectronMVAEstimatorRun2::ElectronMVAEstimatorRun2	(	const std::string &	mvaTag,
		const std::string &	mvaName,
		int	nCategories,
		const std::string &	variableDefinition,
		const std::vector< std::string > &	categoryCutStrings,
		const std::vector< std::string > &	weightFileNames,
		bool	debug = `false`
	)

Definition at line 23 of file ElectronMVAEstimatorRun2.cc.

References categoryFunctions_, TkAlMuonSelectors_cfi::cut, AnyMVAEstimatorRun2Base::getName(), AnyMVAEstimatorRun2Base::getNCategories(), AnyMVAEstimatorRun2Base::getTag(), and init().

   : AnyMVAEstimatorRun2Base( mvaName, mvaTag, nCategories, debug )
   , mvaVarMngr_            (variableDefinition)
 {
   
   if( (int)(categoryCutStrings.size()) != getNCategories() )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of category cuts in " << getName() << getTag() << std::endl;
   
   for (auto const& cut : categoryCutStrings) categoryFunctions_.emplace_back(cut);
   init(weightFileNames);
 }

Member Function Documentation

int ElectronMVAEstimatorRun2::findCategory ( const reco::Candidate * candidate ) const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 125 of file ElectronMVAEstimatorRun2.cc.

References metsig::electron, and Exception.

Referenced by getExtraVars(), and mvaValue().

                                                                                 {
 
   const reco::GsfElectron* electron = dynamic_cast<reco::GsfElectron const*>(candidate);
 
   if( electron == nullptr ) {
     throw cms::Exception("MVA failure: ")
       << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
       << " but appears to be neither" << std::endl;
   }
 
   return findCategory(*electron);
 
 }

int ElectronMVAEstimatorRun2::findCategory ( reco::GsfElectron const & electron ) const

private

Definition at line 139 of file ElectronMVAEstimatorRun2.cc.

References categoryFunctions_, AnyMVAEstimatorRun2Base::getNCategories(), AnyMVAEstimatorRun2Base::getTag(), mps_fire::i, reco::LeafCandidate::pt(), and reco::GsfElectron::superCluster().

                                                                                 {
 
   for (int i = 0; i < getNCategories(); ++i) {
       if (categoryFunctions_[i](electron)) return i;
   }
 
   edm::LogWarning  ("MVA warning") <<
       "category not defined for particle with pt " << electron.pt() << " GeV, eta " <<
           electron.superCluster()->eta() << " in ElectronMVAEstimatorRun2" << getTag();
 
   return -1;
 
 }

static std::vector<float> ElectronMVAEstimatorRun2::getExtraVars	(	reco::GsfElectron const &	ele,
		reco::ConversionCollection const *	conversions,
		reco::BeamSpot const *	beamSpot,
		double	rho
	)

inlinestatic

Definition at line 33 of file ElectronMVAEstimatorRun2.h.

References reco::Vertex::chi2(), reco::GsfElectron::closestCtfTrackRef(), conv, reco::Conversion::conversionVertex(), metsig::electron, findCategory(), init(), edm::Ref< C, T, F >::isAvailable(), edm::Ref< C, T, F >::isNonnull(), reco::Vertex::isValid(), ConversionTools::matchedConversion(), mvaValue(), reco::Vertex::ndof(), reco::BeamSpot::position(), rho, and badGlobalMuonTaggersAOD_cff::vtx.

Referenced by MVAVariableHelper< reco::GsfElectron >::getAuxVariables().

   {
       // Conversion vertex fit
       reco::Conversion const* conv = ConversionTools::matchedConversion(ele, *conversions, beamSpot->position());
 
       float convVtxFitProb = -1.;
       if(!(conv == nullptr)) {
           const reco::Vertex &vtx = conv->conversionVertex();
           if (vtx.isValid()) {
               convVtxFitProb = TMath::Prob( vtx.chi2(),  vtx.ndof());
           }
       }
 
       // kf track related variables
       bool validKf=false;
       reco::TrackRef trackRef = ele.closestCtfTrackRef();
       validKf = trackRef.isAvailable();
       validKf &= trackRef.isNonnull();
       float kfchi2 = validKf ? trackRef->normalizedChi2() : 0 ; //ielectron->track()->normalizedChi2() : 0 ;
       float kfhits = validKf ? trackRef->hitPattern().trackerLayersWithMeasurement() : -1. ;
 
       return std::vector<float>{kfhits, kfchi2, convVtxFitProb, static_cast<float>(rho)};
   }

void ElectronMVAEstimatorRun2::init ( const std::vector< std::string > & weightFileNames )

private

Definition at line 42 of file ElectronMVAEstimatorRun2.cc.

References gather_cfg::cout, createGBRForest(), Exception, gbrForests_, AnyMVAEstimatorRun2Base::getNCategories(), AnyMVAEstimatorRun2Base::getTag(), MVAVariableManager< ParticleType >::getVarIndex(), mps_fire::i, AnyMVAEstimatorRun2Base::isDebug(), mvaValue(), mvaVarMngr_, nVariables_, and variables_.

Referenced by ElectronMVAEstimatorRun2(), and getExtraVars().

                                                                                {
 
   if(isDebug()) {
     std::cout << " *** Inside ElectronMVAEstimatorRun2" << getTag() << std::endl;
   }
 
   // Initialize GBRForests
   if( (int)(weightFileNames.size()) != getNCategories() )
     throw cms::Exception("MVA config failure: ")
       << "wrong number of weightfiles in ElectronMVAEstimatorRun2" << getTag() << std::endl;
 
   gbrForests_.clear();
   // Create a TMVA reader object for each category
   for(int i=0; i<getNCategories(); i++){
 
     std::vector<int> variablesInCategory;
 
     // Use unique_ptr so that all readers are properly cleaned up
     // when the vector clear() is called in the destructor
 
     std::vector<std::string> variableNamesInCategory;
     gbrForests_.push_back(createGBRForest(weightFileNames[i], variableNamesInCategory));
 
     nVariables_.push_back(variableNamesInCategory.size());
 
     variables_.push_back(variablesInCategory);
 
     if(isDebug()) {
       std::cout << " *** Inside ElectronMVAEstimatorRun2" << getTag() << std::endl;
       std::cout << " category " << i << " with nVariables " << nVariables_[i] << std::endl;
     }
 
     for (int j=0; j<nVariables_[i];++j) {
         int index = mvaVarMngr_.getVarIndex(variableNamesInCategory[j]);
         if(index == -1) {
             throw cms::Exception("MVA config failure: ")
                << "Concerning ElectronMVAEstimatorRun2" << getTag() << std::endl
                << "Variable " << variableNamesInCategory[j]
                << " not found in variable definition file!" << std::endl;
         }
         variables_[i].push_back(index);
 
     }
   }
 }

float ElectronMVAEstimatorRun2::mvaValue	(	const reco::Candidate *	candidate,
		std::vector< float > const &	auxVariables,
		int &	iCategory
	)		const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 89 of file ElectronMVAEstimatorRun2.cc.

References gather_cfg::cout, metsig::electron, Exception, findCategory(), gbrForests_, MVAVariableManager< ParticleType >::getName(), AnyMVAEstimatorRun2Base::getTag(), MVAVariableManager< ParticleType >::getValue(), mps_fire::i, AnyMVAEstimatorRun2Base::isDebug(), mvaVarMngr_, nVariables_, and variables_.

Referenced by getExtraVars(), and init().

                                                                                                       {
 
   const reco::GsfElectron* electron = dynamic_cast<const reco::GsfElectron*>(candidate);
 
   if( electron == nullptr ) {
     throw cms::Exception("MVA failure: ")
       << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
       << " but appears to be neither" << std::endl;
   }
 
   iCategory = findCategory( electron );
 
   if (iCategory < 0) return -999;
 
   std::vector<float> vars;
 
   for (int i = 0; i < nVariables_[iCategory]; ++i) {
       vars.push_back(mvaVarMngr_.getValue(variables_[iCategory][i], *electron, auxVariables));
   }
 
   if(isDebug()) {
     std::cout << " *** Inside ElectronMVAEstimatorRun2" << getTag() << std::endl;
     std::cout << " category " << iCategory << std::endl;
     for (int i = 0; i < nVariables_[iCategory]; ++i) {
         std::cout << " " << mvaVarMngr_.getName(variables_[iCategory][i]) << " " << vars[i] << std::endl;
     }
   }
   const float response = gbrForests_.at(iCategory)->GetResponse(vars.data()); // The BDT score
 
   if(isDebug()) {
     std::cout << " ### MVA " << response << std::endl << std::endl;
   }
 
   return response;
 }

Member Data Documentation

std::vector<ThreadSafeStringCut<StringCutObjectSelector<reco::GsfElectron>, reco::GsfElectron> > ElectronMVAEstimatorRun2::categoryFunctions_

private

Definition at line 71 of file ElectronMVAEstimatorRun2.h.

Referenced by ElectronMVAEstimatorRun2(), and findCategory().

std::vector< std::unique_ptr<const GBRForest> > ElectronMVAEstimatorRun2::gbrForests_

private

Definition at line 75 of file ElectronMVAEstimatorRun2.h.

Referenced by init(), and mvaValue().

MVAVariableManager<reco::GsfElectron> ElectronMVAEstimatorRun2::mvaVarMngr_

private

Definition at line 82 of file ElectronMVAEstimatorRun2.h.

Referenced by init(), and mvaValue().

std::vector<int> ElectronMVAEstimatorRun2::nVariables_

private

Definition at line 72 of file ElectronMVAEstimatorRun2.h.

Referenced by init(), and mvaValue().

std::vector<std::vector<int> > ElectronMVAEstimatorRun2::variables_

private

Definition at line 80 of file ElectronMVAEstimatorRun2.h.

Referenced by init(), and mvaValue().

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation