ElectronMVAEstimator Class Reference

#include <ElectronMVAEstimator.h>

Classes
struct	Configuration

Public Member Functions
	ElectronMVAEstimator ()
	ElectronMVAEstimator (std::string fileName)
	ElectronMVAEstimator (const Configuration &)
double	mva (const reco::GsfElectron &myElectron, int nvertices=0) const
	~ElectronMVAEstimator ()

Private Member Functions
void	bindVariables (float vars[18]) const

Private Attributes
Float_t	absdcot
Float_t	absdist
const Configuration	cfg_
Float_t	detaeleout
Float_t	detain
Float_t	dphiin
Float_t	e1x5e5x5
Int_t	ecalseed
Float_t	eleopout
Float_t	eop
Float_t	eta
Float_t	fbrem
std::vector< std::unique_ptr< const GBRForest > >	gbr
Float_t	hoe
Float_t	kfchi2
Float_t	mykfhits
Float_t	mymishits
Float_t	myNvtx
Float_t	pt
Float_t	sieie

Detailed Description

Definition at line 10 of file ElectronMVAEstimator.h.

Constructor & Destructor Documentation

ElectronMVAEstimator::ElectronMVAEstimator

(

)

Definition at line 16 of file ElectronMVAEstimator.cc.

                                          :
  cfg_{}
{}

ElectronMVAEstimator::ElectronMVAEstimator

(

std::string

fileName

)

Definition at line 20 of file ElectronMVAEstimator.cc.

References absdcot, absdist, detaeleout, detain, dphiin, e1x5e5x5, ecalseed, eleopout, eop, eta, fbrem, MillePedeFileConverter_cfg::fileName, gbr, hoe, kfchi2, mykfhits, mymishits, myNvtx, pt, and sieie.

                                                            :
  cfg_{} 
{
  TMVA::Reader tmvaReader("!Color:Silent");
  tmvaReader.AddVariable("fbrem",&fbrem);
  tmvaReader.AddVariable("detain", &detain);
  tmvaReader.AddVariable("dphiin", &dphiin);
  tmvaReader.AddVariable("sieie", &sieie);
  tmvaReader.AddVariable("hoe", &hoe);
  tmvaReader.AddVariable("eop", &eop);
  tmvaReader.AddVariable("e1x5e5x5", &e1x5e5x5);
  tmvaReader.AddVariable("eleopout", &eleopout);
  tmvaReader.AddVariable("detaeleout", &detaeleout);
  tmvaReader.AddVariable("kfchi2", &kfchi2);
  tmvaReader.AddVariable("kfhits", &mykfhits);
  tmvaReader.AddVariable("mishits",&mymishits);
  tmvaReader.AddVariable("dist", &absdist);
  tmvaReader.AddVariable("dcot", &absdcot);
  tmvaReader.AddVariable("nvtx", &myNvtx);

  tmvaReader.AddSpectator("eta",&eta);
  tmvaReader.AddSpectator("pt",&pt);
  tmvaReader.AddSpectator("ecalseed",&ecalseed);
  
  // Taken from Daniele (his mail from the 30/11)
  //  tmvaReader.BookMVA("BDTSimpleCat","../Training/weights_Root527b_3Depth_DanVarConvRej_2PtBins_10Pt_800TPrune5_Min100Events_NoBjets_half/TMVA_BDTSimpleCat.weights.xm");
  // training of the 7/12 with Nvtx added
  tmvaReader.BookMVA(ele_mva_name,fileName.c_str());
  gbr.emplace_back(new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmvaReader.FindMVA(ele_mva_name) ) ) );
}

ElectronMVAEstimator::ElectronMVAEstimator

(

const Configuration &

cfg

)

Definition at line 51 of file ElectronMVAEstimator.cc.

References absdcot, absdist, cfg_, detaeleout, detain, dphiin, e1x5e5x5, ecalseed, eleopout, eop, eta, fbrem, gbr, hoe, compare_using_db::ifile, kfchi2, mykfhits, mymishits, myNvtx, pt, sieie, AlCaHLTBitMon_QueryRunRegistry::string, and ElectronMVAEstimator::Configuration::vweightsfiles.

                                                                   :cfg_(cfg){
  std::vector<std::string> weightsfiles;
  std::string path_mvaWeightFileEleID;
  for(unsigned ifile=0 ; ifile < cfg_.vweightsfiles.size() ; ++ifile) {
    path_mvaWeightFileEleID = edm::FileInPath ( cfg_.vweightsfiles[ifile].c_str() ).fullPath();
    weightsfiles.push_back(path_mvaWeightFileEleID);
  }
  for( const auto& wgtfile : weightsfiles ) {
    TMVA::Reader tmvaReader("!Color:Silent");
    tmvaReader.AddVariable("fbrem",&fbrem);
    tmvaReader.AddVariable("detain", &detain);
    tmvaReader.AddVariable("dphiin", &dphiin);
    tmvaReader.AddVariable("sieie", &sieie);
    tmvaReader.AddVariable("hoe", &hoe);
    tmvaReader.AddVariable("eop", &eop);
    tmvaReader.AddVariable("e1x5e5x5", &e1x5e5x5);
    tmvaReader.AddVariable("eleopout", &eleopout);
    tmvaReader.AddVariable("detaeleout", &detaeleout);
    tmvaReader.AddVariable("kfchi2", &kfchi2);
    tmvaReader.AddVariable("kfhits", &mykfhits);
    tmvaReader.AddVariable("mishits",&mymishits);
    tmvaReader.AddVariable("dist", &absdist);
    tmvaReader.AddVariable("dcot", &absdcot);
    tmvaReader.AddVariable("nvtx", &myNvtx);
    
    tmvaReader.AddSpectator("eta",&eta);
    tmvaReader.AddSpectator("pt",&pt);
    tmvaReader.AddSpectator("ecalseed",&ecalseed);
    
    // Taken from Daniele (his mail from the 30/11)
    //  tmvaReader.BookMVA("BDTSimpleCat","../Training/weights_Root527b_3Depth_DanVarConvRej_2PtBins_10Pt_800TPrune5_Min100Events_NoBjets_half/TMVA_BDTSimpleCat.weights.xm");
    // training of the 7/12 with Nvtx added
    tmvaReader.BookMVA(ele_mva_name,wgtfile);
    gbr.emplace_back(new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmvaReader.FindMVA(ele_mva_name) ) ) );
  }
}

ElectronMVAEstimator::~ElectronMVAEstimator ( )

inline

Definition at line 18 of file ElectronMVAEstimator.h.

References mva().

{;}

Member Function Documentation

void ElectronMVAEstimator::bindVariables ( float  vars[18] ) const

private

Definition at line 151 of file ElectronMVAEstimator.cc.

Referenced by mva().

                                                             {
  if(vars[0] < -1.)
    vars[1] = -1.;  
  
  if(vars[1] > 0.06)
    vars[1] = 0.06;
    
  if(vars[2] > 0.6)
    vars[2] = 0.6;
  
  if(vars[5] > 20.)
    vars[5] = 20.;
    
  if(vars[7] > 20.)
    vars[7] = 20;
  
  if(vars[8] > 0.2)
    vars[8] = 0.2;
  
  if(vars[9] < 0.)
    vars[9] = 0.;
  
  if(vars[9] > 15.)
    vars[9] = 15.;
    
  if(vars[6] < -1.)
    vars[6] = -1;

  if(vars[6] > 2.)
    vars[6] = 2.; 
    
  if(vars[12] > 15.)
    vars[12] = 15.;
    
  if(vars[13] > 3.)
    vars[13] = 3.;
}

double ElectronMVAEstimator::mva	(	const reco::GsfElectron &	myElectron,
		int	nvertices = 0
	)		const

Definition at line 88 of file ElectronMVAEstimator.cc.

References funct::abs(), bindVariables(), reco::GsfElectron::closestCtfTrackRef(), reco::GsfElectron::convDcot(), reco::GsfElectron::convDist(), reco::GsfElectron::deltaEtaEleClusterTrackAtCalo(), reco::GsfElectron::deltaEtaSuperClusterTrackAtVtx(), reco::GsfElectron::deltaPhiSuperClusterTrackAtVtx(), reco::GsfElectron::e1x5(), reco::GsfElectron::e5x5(), reco::GsfElectron::ecalDrivenSeed(), reco::GsfElectron::eEleClusterOverPout(), reco::GsfElectron::eSuperClusterOverP(), reco::LeafCandidate::eta(), reco::GsfElectron::fbrem(), gbr, reco::GsfElectron::gsfTrack(), reco::GsfElectron::hcalOverEcal(), diffTreeTool::index, edm::Ref< C, T, F >::isAvailable(), edm::Ref< C, T, F >::isNonnull(), reco::HitPattern::MISSING_INNER_HITS, reco::LeafCandidate::pt(), mps_fire::result, and reco::GsfElectron::sigmaIetaIeta().

Referenced by ~ElectronMVAEstimator().

                                                                                        {
  float vars[18];

  vars[0] = myElectron.fbrem();
  vars[1] = std::abs(myElectron.deltaEtaSuperClusterTrackAtVtx());
  vars[2] = std::abs(myElectron.deltaPhiSuperClusterTrackAtVtx());
  vars[3] = myElectron.sigmaIetaIeta();
  vars[4] = myElectron.hcalOverEcal();
  vars[5] = myElectron.eSuperClusterOverP();
  vars[6] = (myElectron.e5x5()) !=0. ? 1.-(myElectron.e1x5()/myElectron.e5x5()) : -1. ;
  vars[7] = myElectron.eEleClusterOverPout();
  vars[8] = std::abs(myElectron.deltaEtaEleClusterTrackAtCalo());
  
  bool validKF= false;

  reco::TrackRef myTrackRef = myElectron.closestCtfTrackRef();
  validKF = (myTrackRef.isNonnull() && myTrackRef.isAvailable());   

  vars[9] = (validKF) ? myTrackRef->normalizedChi2() : 0 ;
  vars[10] = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1.; 
  vars[11] = myElectron.gsfTrack()->hitPattern().numberOfLostHits(reco::HitPattern::MISSING_INNER_HITS);
  vars[12] = std::abs(myElectron.convDist());
  vars[13] = std::abs(myElectron.convDcot());
  vars[14] = nvertices;
  vars[15] = myElectron.eta();
  vars[16] = myElectron.pt();
  vars[17] = myElectron.ecalDrivenSeed();
  
  bindVariables(vars);

  //0 pt < 10 && abs(eta)<=1.485
  //1 pt >= 10 && abs(eta)<=1.485
  //2 pt < 10 && abs(eta)> 1.485
  //3 pt >= 10 &&  abs(eta)> 1.485

  const unsigned index = (unsigned)(myElectron.pt() >= 10) + 2*(unsigned)(std::abs(myElectron.eta()) > 1.485);

  double result =  gbr[index]->GetAdaBoostClassifier(vars);
//  
//  std::cout << "fbrem" << vars[0] << std::endl;
//  std::cout << "detain"<< vars[1] << std::endl;
//  std::cout << "dphiin"<< vars[2] << std::endl;
//  std::cout << "sieie"<< vars[3] << std::endl;
//  std::cout << "hoe"<< vars[4] << std::endl;
//  std::cout << "eop"<< vars[5] << std::endl;
//  std::cout << "e1x5e5x5"<< vars[6] << std::endl;
//  std::cout << "eleopout"<< vars[7] << std::endl;
//  std::cout << "detaeleout"<< vars[8] << std::endl;
//  std::cout << "kfchi2"<< vars[9] << std::endl;
//  std::cout << "kfhits"<< vars[10] << std::endl;
//  std::cout << "mishits"<<vars[11] << std::endl;
//  std::cout << "dist"<< vars[12] << std::endl;
//  std::cout << "dcot"<< vars[13] << std::endl;
//  std::cout << "nvtx"<< vars[14] << std::endl;
//  std::cout << "eta"<< vars[15] << std::endl;
//  std::cout << "pt"<< vars[16] << std::endl;
//  std::cout << "ecalseed"<< vars[17] << std::endl;
//
//  std::cout << " MVA " << result << std::endl;
  return result;
}

Member Data Documentation

Float_t ElectronMVAEstimator::absdcot

private

Definition at line 40 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::absdist

private

Definition at line 39 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

const Configuration ElectronMVAEstimator::cfg_

private

Definition at line 22 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::detaeleout

private

Definition at line 35 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::detain

private

Definition at line 28 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::dphiin

private

Definition at line 29 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::e1x5e5x5

private

Definition at line 33 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Int_t ElectronMVAEstimator::ecalseed

private

Definition at line 44 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::eleopout

private

Definition at line 34 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::eop

private

Definition at line 32 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::eta

private

Definition at line 42 of file ElectronMVAEstimator.h.

Referenced by Particle.Particle::__str__(), ElectronMVAEstimator(), Jet.Jet::jetID(), and Jet.Jet::puJetId().

Float_t ElectronMVAEstimator::fbrem

private

Definition at line 27 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

std::vector<std::unique_ptr<const GBRForest> > ElectronMVAEstimator::gbr

private

Definition at line 25 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator(), and mva().

Float_t ElectronMVAEstimator::hoe

private

Definition at line 31 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::kfchi2

private

Definition at line 36 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::mykfhits

private

Definition at line 37 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::mymishits

private

Definition at line 38 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::myNvtx

private

Definition at line 41 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().

Float_t ElectronMVAEstimator::pt

private

Definition at line 43 of file ElectronMVAEstimator.h.

Referenced by Particle.Particle::__str__(), ZMuMuRochCorAnalyzer.DiMuon::__str__(), DiObject.DiMuon::__str__(), Photon.Photon::calScaledIsoValueExp(), Photon.Photon::calScaledIsoValueLin(), Photon.Photon::calScaledIsoValueQuadr(), ElectronMVAEstimator(), Electron.Electron::mvaIDLoose(), Electron.Electron::mvaIDRun2(), Electron.Electron::mvaIDTight(), Electron.Electron::ptErr(), ntupleDataFormat.Track::ptPull(), Lepton.Lepton::relIso(), Lepton.Lepton::relIsoFromEA(), Lepton.Lepton::relIsoR(), and Jet.Jet::setCorrP4().

Float_t ElectronMVAEstimator::sieie

private

Definition at line 30 of file ElectronMVAEstimator.h.

Referenced by ElectronMVAEstimator().