CandidateBoostedDoubleSecondaryVertexComputer Class Reference

#include <CandidateBoostedDoubleSecondaryVertexComputer.h>

Inheritance diagram for CandidateBoostedDoubleSecondaryVertexComputer:

Classes
struct	Tokens

Public Member Functions
	CandidateBoostedDoubleSecondaryVertexComputer (const edm::ParameterSet &parameters, Tokens tokens)
float	discriminator (const TagInfoHelper &tagInfos) const override
void	initialize (const JetTagComputerRecord &) override
Public Member Functions inherited from JetTagComputer
const std::vector< std::string > &	getInputLabels () const
	JetTagComputer ()
	JetTagComputer (const edm::ParameterSet &configuration)
float	operator() (const reco::BaseTagInfo &info) const
float	operator() (const TagInfoHelper &helper) const
void	setupDone ()
virtual	~JetTagComputer ()

Private Attributes
std::unique_ptr< TMVAEvaluator >	mvaID
const Tokens	tokens_
const bool	useAdaBoost_
const bool	useGBRForest_
const edm::FileInPath	weightFile_

Additional Inherited Members
Protected Member Functions inherited from JetTagComputer
virtual float	discriminator (const reco::BaseTagInfo &) const
void	uses (const std::string &label)
void	uses (unsigned int id, const std::string &label)

Detailed Description

Definition at line 10 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Constructor & Destructor Documentation

CandidateBoostedDoubleSecondaryVertexComputer()

CandidateBoostedDoubleSecondaryVertexComputer::CandidateBoostedDoubleSecondaryVertexComputer	(	const edm::ParameterSet &	parameters,
		Tokens	tokens
	)

Definition at line 20 of file CandidateBoostedDoubleSecondaryVertexComputer.cc.

    : weightFile_(parameters.existsAs<edm::FileInPath>("weightFile")
                      ? parameters.getParameter<edm::FileInPath>("weightFile")
                      : edm::FileInPath()),
      useGBRForest_(parameters.existsAs<bool>("useGBRForest") ? parameters.getParameter<bool>("useGBRForest") : false),
      useAdaBoost_(parameters.existsAs<bool>("useAdaBoost") ? parameters.getParameter<bool>("useAdaBoost") : false),
      tokens_{std::move(tokens)} {
  uses(0, "svTagInfos");
 
  mvaID.reset(new TMVAEvaluator());
}

References eostools::move().

Member Function Documentation

discriminator()

float CandidateBoostedDoubleSecondaryVertexComputer::discriminator ( const TagInfoHelper &  tagInfos ) const

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 72 of file CandidateBoostedDoubleSecondaryVertexComputer.cc.

                                                                                                     {
  // get the TagInfo
  const reco::BoostedDoubleSVTagInfo& bdsvTagInfo = tagInfo.get<reco::BoostedDoubleSVTagInfo>(0);
 
  // get the TaggingVariables
  const reco::TaggingVariableList vars = bdsvTagInfo.taggingVariables();
 
  // default discriminator value
  float value = -10.;
 
  std::map<std::string, float> inputs;
  inputs["z_ratio"] = vars.get(reco::btau::z_ratio);
  inputs["trackSipdSig_3"] = vars.get(reco::btau::trackSip3dSig_3);
  inputs["trackSipdSig_2"] = vars.get(reco::btau::trackSip3dSig_2);
  inputs["trackSipdSig_1"] = vars.get(reco::btau::trackSip3dSig_1);
  inputs["trackSipdSig_0"] = vars.get(reco::btau::trackSip3dSig_0);
  inputs["trackSipdSig_1_0"] = vars.get(reco::btau::tau2_trackSip3dSig_0);
  inputs["trackSipdSig_0_0"] = vars.get(reco::btau::tau1_trackSip3dSig_0);
  inputs["trackSipdSig_1_1"] = vars.get(reco::btau::tau2_trackSip3dSig_1);
  inputs["trackSipdSig_0_1"] = vars.get(reco::btau::tau1_trackSip3dSig_1);
  inputs["trackSip2dSigAboveCharm_0"] = vars.get(reco::btau::trackSip2dSigAboveCharm);
  inputs["trackSip2dSigAboveBottom_0"] = vars.get(reco::btau::trackSip2dSigAboveBottom_0);
  inputs["trackSip2dSigAboveBottom_1"] = vars.get(reco::btau::trackSip2dSigAboveBottom_1);
  inputs["tau1_trackEtaRel_0"] = vars.get(reco::btau::tau2_trackEtaRel_0);
  inputs["tau1_trackEtaRel_1"] = vars.get(reco::btau::tau2_trackEtaRel_1);
  inputs["tau1_trackEtaRel_2"] = vars.get(reco::btau::tau2_trackEtaRel_2);
  inputs["tau0_trackEtaRel_0"] = vars.get(reco::btau::tau1_trackEtaRel_0);
  inputs["tau0_trackEtaRel_1"] = vars.get(reco::btau::tau1_trackEtaRel_1);
  inputs["tau0_trackEtaRel_2"] = vars.get(reco::btau::tau1_trackEtaRel_2);
  inputs["tau_vertexMass_0"] = vars.get(reco::btau::tau1_vertexMass);
  inputs["tau_vertexEnergyRatio_0"] = vars.get(reco::btau::tau1_vertexEnergyRatio);
  inputs["tau_vertexDeltaR_0"] = vars.get(reco::btau::tau1_vertexDeltaR);
  inputs["tau_flightDistance2dSig_0"] = vars.get(reco::btau::tau1_flightDistance2dSig);
  inputs["tau_vertexMass_1"] = vars.get(reco::btau::tau2_vertexMass);
  inputs["tau_vertexEnergyRatio_1"] = vars.get(reco::btau::tau2_vertexEnergyRatio);
  inputs["tau_flightDistance2dSig_1"] = vars.get(reco::btau::tau2_flightDistance2dSig);
  inputs["jetNTracks"] = vars.get(reco::btau::jetNTracks);
  inputs["nSV"] = vars.get(reco::btau::jetNSecondaryVertices);
 
  // evaluate the MVA
  value = mvaID->evaluate(inputs);
 
  // return the final discriminator value
  return value;
}

References PixelMapPlotter::inputs, reco::btau::jetNSecondaryVertices, reco::btau::jetNTracks, mvaID, reco::BoostedDoubleSVTagInfo::taggingVariables(), dumpRecoGeometry_cfg::tagInfo, reco::btau::tau1_flightDistance2dSig, reco::btau::tau1_trackEtaRel_0, reco::btau::tau1_trackEtaRel_1, reco::btau::tau1_trackEtaRel_2, reco::btau::tau1_trackSip3dSig_0, reco::btau::tau1_trackSip3dSig_1, reco::btau::tau1_vertexDeltaR, reco::btau::tau1_vertexEnergyRatio, reco::btau::tau1_vertexMass, reco::btau::tau2_flightDistance2dSig, reco::btau::tau2_trackEtaRel_0, reco::btau::tau2_trackEtaRel_1, reco::btau::tau2_trackEtaRel_2, reco::btau::tau2_trackSip3dSig_0, reco::btau::tau2_trackSip3dSig_1, reco::btau::tau2_vertexEnergyRatio, reco::btau::tau2_vertexMass, reco::btau::trackSip2dSigAboveBottom_0, reco::btau::trackSip2dSigAboveBottom_1, reco::btau::trackSip2dSigAboveCharm, reco::btau::trackSip3dSig_0, reco::btau::trackSip3dSig_1, reco::btau::trackSip3dSig_2, reco::btau::trackSip3dSig_3, relativeConstraints::value, and reco::btau::z_ratio.

initialize()

void CandidateBoostedDoubleSecondaryVertexComputer::initialize ( const JetTagComputerRecord &  record )

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 33 of file CandidateBoostedDoubleSecondaryVertexComputer.cc.

                                                                                                 {
  // variable names and order need to be the same as in the training
  std::vector<std::string> variables({"z_ratio",
                                      "trackSipdSig_3",
                                      "trackSipdSig_2",
                                      "trackSipdSig_1",
                                      "trackSipdSig_0",
                                      "trackSipdSig_1_0",
                                      "trackSipdSig_0_0",
                                      "trackSipdSig_1_1",
                                      "trackSipdSig_0_1",
                                      "trackSip2dSigAboveCharm_0",
                                      "trackSip2dSigAboveBottom_0",
                                      "trackSip2dSigAboveBottom_1",
                                      "tau0_trackEtaRel_0",
                                      "tau0_trackEtaRel_1",
                                      "tau0_trackEtaRel_2",
                                      "tau1_trackEtaRel_0",
                                      "tau1_trackEtaRel_1",
                                      "tau1_trackEtaRel_2",
                                      "tau_vertexMass_0",
                                      "tau_vertexEnergyRatio_0",
                                      "tau_vertexDeltaR_0",
                                      "tau_flightDistance2dSig_0",
                                      "tau_vertexMass_1",
                                      "tau_vertexEnergyRatio_1",
                                      "tau_flightDistance2dSig_1",
                                      "jetNTracks",
                                      "nSV"});
  // book TMVA readers
  std::vector<std::string> spectators({"massPruned", "flavour", "nbHadrons", "ptPruned", "etaPruned"});
 
  if (tokens_.gbrForest_.isInitialized()) {
    mvaID->initializeGBRForest(&record.get(tokens_.gbrForest_), variables, spectators, useAdaBoost_);
  } else
    mvaID->initialize(
        "Color:Silent:Error", "BDT", weightFile_.fullPath(), variables, spectators, useGBRForest_, useAdaBoost_);
}

References edm::FileInPath::fullPath(), CandidateBoostedDoubleSecondaryVertexComputer::Tokens::gbrForest_, edm::ESGetToken< ESProduct, ESRecord >::isInitialized(), mvaID, GlobalPosition_Frontier_DevDB_cff::record, candidateCombinedMVAV2Computer_cfi::spectators, tokens_, useAdaBoost_, useGBRForest_, L1TEGammaDiff_cfi::variables, and weightFile_.

Member Data Documentation

mvaID

std::unique_ptr<TMVAEvaluator> CandidateBoostedDoubleSecondaryVertexComputer::mvaID

private

Definition at line 28 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Referenced by discriminator(), and initialize().

tokens_

const Tokens CandidateBoostedDoubleSecondaryVertexComputer::tokens_

private

Definition at line 26 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Referenced by initialize().

useAdaBoost_

const bool CandidateBoostedDoubleSecondaryVertexComputer::useAdaBoost_

private

Definition at line 25 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Referenced by initialize().

useGBRForest_

const bool CandidateBoostedDoubleSecondaryVertexComputer::useGBRForest_

private

Definition at line 24 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Referenced by initialize().

weightFile_

const edm::FileInPath CandidateBoostedDoubleSecondaryVertexComputer::weightFile_

private

Definition at line 23 of file CandidateBoostedDoubleSecondaryVertexComputer.h.

Referenced by initialize().