#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 constexpr 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 12 of file SoftElectronMVAEstimator.h.

Constructor & Destructor Documentation

SoftElectronMVAEstimator::SoftElectronMVAEstimator ( const Configuration & cfg )

Definition at line 9 of file SoftElectronMVAEstimator.cc.

References cfg_, 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(createGBRForest(weightsfile));
   }
 }

SoftElectronMVAEstimator::~SoftElectronMVAEstimator ( )

Definition at line 24 of file SoftElectronMVAEstimator.cc.

24 {}

Member Function Documentation

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

private

Definition at line 95 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 26 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;
 }