#include <SoftElectronMVAEstimator.h>

Classes
struct	Configuration

Public Member Functions
double	mva (const reco::GsfElectron &myElectron, const reco::VertexCollection &) const

	SoftElectronMVAEstimator (const Configuration &)

	~SoftElectronMVAEstimator ()

Static Public Attributes
static unsigned int	ExpectedNBins = 1

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

void	init ()

Private Attributes
const Configuration	cfg_

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

Detailed Description

Definition at line 14 of file SoftElectronMVAEstimator.h.

Constructor & Destructor Documentation

SoftElectronMVAEstimator::SoftElectronMVAEstimator ( const Configuration & cfg )

Definition at line 7 of file SoftElectronMVAEstimator.cc.

References cfg_, GBRForestTools::createGBRForest(), ExpectedNBins, gbr_, and SoftElectronMVAEstimator::Configuration::vweightsfiles.

                                                                            :cfg_(cfg)
 {
   //Check number of weight files given
   if (ExpectedNBins != cfg_.vweightsfiles.size() ) {
     edm::LogError  ("Soft Electron MVA Error") <<
         "Expected Number of bins = " << ExpectedNBins <<
         " does not equal to weightsfiles.size() = " << 
         cfg_.vweightsfiles.size() << std::endl;
   }
 
   for(auto& weightsfile : cfg_.vweightsfiles) {
     // Taken from Daniele (his mail from the 30/11)    
     // training of the 7/12 with Nvtx added
     gbr_.push_back( GBRForestTools::createGBRForest( weightsfile ));
   }
 }

SoftElectronMVAEstimator::~SoftElectronMVAEstimator ( )

Definition at line 25 of file SoftElectronMVAEstimator.cc.

26 { }

Member Function Documentation

void SoftElectronMVAEstimator::bindVariables ( float vars[25] ) const

private

Definition at line 99 of file SoftElectronMVAEstimator.cc.

References funct::abs().

Referenced by mva().

                                                                  {
   if( vars[0] < -1.) //fbrem
     vars[0] = -1.;
 
   vars[11] = std::abs(vars[11]); // deta
   if(vars[11] > 0.06)
     vars[11] = 0.06;
 
   vars[12] = std::abs(vars[12]);
   if(vars[12] > 0.6)
     vars[12] = 0.6;
 
   //if(EoP > 20.)
   //  EoP = 20.;
 
   if(vars[2] > 20.) //eleEoPout
     vars[2] = 20.;
   
   vars[13] = std::abs(vars[13]); //detacalo
   if(vars[13] > 0.2)
     vars[13] = 0.2;
 
   if( vars[19] < -1.) //OneMinusE1x5E5x5
     vars[19] = -1;
 
   if( vars[19] > 2.) //OneMinusE1x5E5x5
     vars[19] = 2.;
 
   if( vars[7] > 200.) //gsfchi2
     vars[7] = 200;
   
   if( vars[8] > 10.) //kfchi2
     vars[8]  = 10.;
 
 }

void SoftElectronMVAEstimator::init ( )

private

double SoftElectronMVAEstimator::mva	(	const reco::GsfElectron &	myElectron,
		const reco::VertexCollection &	pvc
	)		const

Definition at line 28 of file SoftElectronMVAEstimator.cc.

                                                                             {
   float vars[25];
 
   vars[0]  = myElectron.fbrem();// fbrem
   vars[1]  = myElectron.eSuperClusterOverP(); //EtotOvePin
   vars[2]   = myElectron.eEleClusterOverPout(); //eleEoPout 
   
   float etot  = myElectron.eSuperClusterOverP()*myElectron.trackMomentumAtVtx().R();
   float eEcal = myElectron.eEleClusterOverPout()*myElectron.trackMomentumAtEleClus().R();
   float dP    = myElectron.trackMomentumAtVtx().R()-myElectron.trackMomentumAtEleClus().R();
   vars[3]  = (etot-eEcal)/dP; //EBremOverDeltaP
   vars[4]  = std::log(myElectron.sigmaEtaEta()); //logSigmaEtaEta
   vars[5]  = myElectron.deltaEtaEleClusterTrackAtCalo(); //DeltaEtaTrackEcalSeed
   vars[6]  = myElectron.hcalOverEcalBc(); //HoE  
 
   bool validKF= false;
   reco::TrackRef myTrackRef     = myElectron.closestCtfTrackRef();
   validKF                       = (myTrackRef.isAvailable() && myTrackRef.isNonnull());
   vars[7]  = myElectron.gsfTrack()->normalizedChi2(); //gsfchi2
   vars[8]  = (validKF) ? myTrackRef->normalizedChi2() : 0 ; //kfchi2
   vars[9]  = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1. ; //kfhits
   
   vars[10] = myElectron.gsfTrack().get()->ptModeError()/myElectron.gsfTrack().get()->ptMode() ; //SigmaPtOverPt 
   vars[11]  = myElectron.deltaEtaSuperClusterTrackAtVtx(); //deta
   vars[12]  = myElectron.deltaPhiSuperClusterTrackAtVtx(); //dphi
   vars[13]  = myElectron.deltaEtaSeedClusterTrackAtCalo(); //detacalo 
   vars[14]  = myElectron.sigmaIetaIeta(); //see
   vars[15]  = myElectron.sigmaIphiIphi(); //spp
   vars[16]  = myElectron.r9(); //R9
   vars[17]  = myElectron.superCluster()->etaWidth(); //etawidth
   vars[18]  = myElectron.superCluster()->phiWidth(); //phiwidth
   vars[19]  = (myElectron.e5x5()) !=0. ? 1.-(myElectron.e1x5()/myElectron.e5x5()) : -1. ; //OneMinusE1x5E5x5
   vars[20]  = (1.0/myElectron.ecalEnergy()) - (1.0 / myElectron.p()); // IoEmIoP
   vars[21]  = myElectron.superCluster()->preshowerEnergy() / myElectron.superCluster()->rawEnergy(); //PreShowerOverRaw
   vars[22]  = pvc.size(); // nPV
   vars[23]  = myElectron.pt(); //pt
   vars[24]  = myElectron.eta(); //eta
   
 /*
   std::cout<<"fbrem "<<fbrem<<std::endl;
   std::cout<<"EtotOvePin "<<EtotOvePin<<std::endl;
   std::cout<<"eleEoPout "<<eleEoPout<<std::endl;
   std::cout<<"EBremOverDeltaP "<<EBremOverDeltaP<<std::endl;
   std::cout<<"logSigmaEtaEta "<<logSigmaEtaEta<<std::endl;
   std::cout<<"DeltaEtaTrackEcalSeed "<<DeltaEtaTrackEcalSeed<<std::endl;
   std::cout<<"HoE "<<HoE<<std::endl;
   std::cout<<"kfchi2 "<<kfchi2<<std::endl;
   std::cout<<"kfhits "<<kfhits<<std::endl;
   std::cout<<"gsfchi2 "<<gsfchi2<<std::endl;
   std::cout<<"SigmaPtOverPt "<<SigmaPtOverPt<<std::endl;
   std::cout<<"deta "<<deta<<std::endl;
   std::cout<<"dphi "<<dphi<<std::endl;
   std::cout<<"detacalo "<<detacalo<<std::endl;
   std::cout<<"see "<<see<<std::endl;
   std::cout<< "spp "             <<          spp<< std::endl;
   std::cout<< "R9 "             <<         R9<< std::endl;
   std::cout<< "IoEmIoP "        <<         IoEmIoP<< std::endl;
   std::cout<<"etawidth "<<etawidth<<std::endl;
   std::cout<<"phiwidth "<<phiwidth<<std::endl;
   std::cout<<"OneMinusE1x5E5x5 "<<OneMinusE1x5E5x5<<std::endl;
   std::cout<<"PreShowerOverRaw "<<PreShowerOverRaw<<std::endl;
 */
   bindVariables(vars);
 
   double result= gbr_[0]->GetClassifier(vars);
 
   return result;
 }