#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 (unsigned int id, const std::string &label)

void	uses (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 21 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_{tokens} {
   uses(0, "svTagInfos");
 
   mvaID = std::make_unique<TMVAEvaluator>();
 }

Member Function Documentation

◆ discriminator()

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

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 73 of file CandidateBoostedDoubleSecondaryVertexComputer.cc.

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.

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

◆ initialize()

void CandidateBoostedDoubleSecondaryVertexComputer::initialize ( const JetTagComputerRecord & record )

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 34 of file CandidateBoostedDoubleSecondaryVertexComputer.cc.

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

                                                                                                  {
   // 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_);
 }