SoftElectronMVAEstimator Class Reference

#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::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::~SoftElectronMVAEstimator

(

)

Definition at line 24 of file SoftElectronMVAEstimator.cc.

{}

Member Function Documentation

bindVariables()

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.;
}

init()

void SoftElectronMVAEstimator::init ( )

private

mva()

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

Definition at line 26 of file SoftElectronMVAEstimator.cc.

References bindVariables(), reco::GsfElectron::closestCtfTrackRef(), reco::GsfElectron::deltaEtaEleClusterTrackAtCalo(), reco::GsfElectron::deltaEtaSeedClusterTrackAtCalo(), reco::GsfElectron::deltaEtaSuperClusterTrackAtVtx(), reco::GsfElectron::deltaPhiSuperClusterTrackAtVtx(), reco::GsfElectron::e1x5(), reco::GsfElectron::e5x5(), reco::GsfElectron::ecalEnergy(), reco::GsfElectron::eEleClusterOverPout(), reco::GsfElectron::eSuperClusterOverP(), reco::LeafCandidate::eta(), reco::GsfElectron::fbrem(), gbr_, edm::Ref< C, T, F >::get(), reco::GsfElectron::gsfTrack(), reco::GsfElectron::hcalOverEcalBc(), edm::Ref< C, T, F >::isAvailable(), edm::Ref< C, T, F >::isNonnull(), CrabHelper::log, reco::LeafCandidate::p(), reco::LeafCandidate::pt(), reco::GsfElectron::r9(), mps_fire::result, reco::GsfElectron::sigmaEtaEta(), reco::GsfElectron::sigmaIetaIeta(), reco::GsfElectron::sigmaIphiIphi(), reco::GsfElectron::superCluster(), reco::GsfElectron::trackMomentumAtEleClus(), and reco::GsfElectron::trackMomentumAtVtx().

                                                                                                             {
  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;
}

Member Data Documentation

cfg_

const Configuration SoftElectronMVAEstimator::cfg_

private

Definition at line 28 of file SoftElectronMVAEstimator.h.

Referenced by SoftElectronMVAEstimator().

ExpectedNBins

constexpr unsigned int SoftElectronMVAEstimator::ExpectedNBins = 1

static

Definition at line 14 of file SoftElectronMVAEstimator.h.

Referenced by SoftElectronMVAEstimator().

gbr_

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

private

Definition at line 29 of file SoftElectronMVAEstimator.h.

Referenced by mva(), and SoftElectronMVAEstimator().