CharmTagger Class Reference

#include <CharmTagger.h>

Inheritance diagram for CharmTagger:

Classes
struct	MVAVar

Public Types
typedef std::vector < edm::ParameterSet >	vpset

Public Member Functions
	CharmTagger (const edm::ParameterSet &)
	explicit ctor More...
virtual float	discriminator (const TagInfoHelper &tagInfo) const override
	b-tag a jet based on track-to-jet parameters in the extened info collection More...
virtual void	initialize (const JetTagComputerRecord &record) override
	~CharmTagger ()
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::string	gbrForest_label_
std::string	mva_name_
std::unique_ptr< TMVAEvaluator >	mvaID_
CombinedSVSoftLeptonComputer	sl_computer_
bool	use_adaBoost_
bool	use_condDB_
bool	use_GBRForest_
std::vector< MVAVar >	variables_
edm::FileInPath	weight_file_

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

Author

M. Verzetti, U. Rochester, N.Y. copied from ElectronTagger.h

Definition at line 20 of file CharmTagger.h.

Member Typedef Documentation

typedef std::vector<edm::ParameterSet> CharmTagger::vpset

Definition at line 28 of file CharmTagger.h.

Constructor & Destructor Documentation

CharmTagger::CharmTagger

(

const edm::ParameterSet &

configuration

)

explicit ctor

Definition at line 15 of file CharmTagger.cc.

References CharmTagger::MVAVar::default_value, edm::ParameterSet::getParameter(), reco::getTaggingVariableName(), CharmTagger::MVAVar::has_index, CharmTagger::MVAVar::id, CharmTagger::MVAVar::index, CharmTagger::MVAVar::name, AlCaHLTBitMon_QueryRunRegistry::string, JetTagComputer::uses(), MetTreeProducer::var(), and variables_.

                                                             :
    sl_computer_(configuration.getParameter<edm::ParameterSet>("slComputerCfg")),
  mva_name_( configuration.getParameter<std::string >("mvaName") ),
  use_condDB_(configuration.getParameter<bool>("useCondDB")),
  gbrForest_label_(configuration.getParameter<std::string>("gbrForestLabel")),
  weight_file_(configuration.getParameter<edm::FileInPath>("weightFile")),
  use_GBRForest_(configuration.getParameter<bool>("useGBRForest")),
  use_adaBoost_(configuration.getParameter<bool>("useAdaBoost"))
{
    vpset vars_definition = configuration.getParameter<vpset>("variables");
    for(auto &var : vars_definition) {
        MVAVar mva_var;
        mva_var.name = var.getParameter<std::string>("name");
        mva_var.id = reco::getTaggingVariableName(
            var.getParameter<std::string>("taggingVarName")
            );
        mva_var.has_index = var.existsAs<int>("idx") ;
        mva_var.index = mva_var.has_index ? var.getParameter<int>("idx") : 0;
        mva_var.default_value = var.getParameter<double>("default");

        variables_.push_back(mva_var);
    }

    uses(0, "pfImpactParameterTagInfos");
    uses(1, "pfInclusiveSecondaryVertexFinderCvsLTagInfos");
    uses(2, "softPFMuonsTagInfos");
    uses(3, "softPFElectronsTagInfos");
}

CharmTagger::~CharmTagger

(

)

Definition at line 76 of file CharmTagger.cc.

{
}

Member Function Documentation

float CharmTagger::discriminator ( const TagInfoHelper &  tagInfo ) const

overridevirtual

b-tag a jet based on track-to-jet parameters in the extened info collection

Reimplemented from JetTagComputer.

Definition at line 81 of file CharmTagger.cc.

References JetTagComputer::TagInfoHelper::get(), SiPixelRawToDigiRegional_cfi::inputs, mvaID_, sl_computer_, GlobalPosition_Frontier_DevDB_cff::tag, create_public_pileup_plots::vals, and variables_.

                                                                    {
  // default value, used if there are no leptons associated to this jet
  const reco::CandIPTagInfo & ip_info = tagInfo.get<reco::CandIPTagInfo>(0);
    const reco::CandSecondaryVertexTagInfo & sv_info = tagInfo.get<reco::CandSecondaryVertexTagInfo>(1);
    const reco::CandSoftLeptonTagInfo& softmu_info = tagInfo.get<reco::CandSoftLeptonTagInfo>(2);
    const reco::CandSoftLeptonTagInfo& softel_info = tagInfo.get<reco::CandSoftLeptonTagInfo>(3);
    reco::TaggingVariableList vars = sl_computer_(ip_info, sv_info, softmu_info, softel_info);

    // Loop over input variables
    std::map<std::string, float> inputs;
    for(auto &mva_var : variables_){
        //vectorial tagging variable
        if(mva_var.has_index){
            std::vector<float> vals = vars.getList(mva_var.id, false);
            inputs[mva_var.name] = (vals.size() > mva_var.index) ? vals[mva_var.index] : mva_var.default_value;
        }
        //single value tagging var
        else {
            inputs[mva_var.name] = vars.get(mva_var.id, mva_var.default_value);
        }
    }

    //get the MVA output
    float tag = mvaID_->evaluate(inputs);
    return tag;
}

void CharmTagger::initialize ( const JetTagComputerRecord &  record )

overridevirtual

Reimplemented from JetTagComputer.

Definition at line 44 of file CharmTagger.cc.

References edm::FileInPath::fullPath(), gbrForest_label_, edm::eventsetup::EventSetupRecord::get(), edm::eventsetup::DependentRecordImplementation< RecordT, ListT >::getRecord(), mva_name_, mvaID_, use_adaBoost_, use_condDB_, use_GBRForest_, MetTreeProducer::var(), variables_, and weight_file_.

{
    mvaID_.reset(new TMVAEvaluator());

    std::vector<std::string> variable_names;
    variable_names.reserve(variables_.size());

    for(auto &var : variables_) {
        variable_names.push_back(var.name);
    }
    std::vector<std::string> spectators;

  if(use_condDB_) {
        const GBRWrapperRcd & gbrWrapperRecord = record.getRecord<GBRWrapperRcd>();

        edm::ESHandle<GBRForest> gbrForestHandle;
        gbrWrapperRecord.get(gbrForest_label_.c_str(), gbrForestHandle);

        mvaID_->initializeGBRForest(
            gbrForestHandle.product(), variable_names, 
            spectators, use_adaBoost_
            );
  }
  else {
    mvaID_->initialize(
            "Color:Silent:Error", mva_name_.c_str(),
            weight_file_.fullPath(), variable_names, 
            spectators, use_GBRForest_, use_adaBoost_
            );
  }
}

Member Data Documentation

std::string CharmTagger::gbrForest_label_

private

Definition at line 45 of file CharmTagger.h.

Referenced by initialize().

std::string CharmTagger::mva_name_

private

Definition at line 43 of file CharmTagger.h.

Referenced by initialize().

std::unique_ptr<TMVAEvaluator> CharmTagger::mvaID_

private

Definition at line 39 of file CharmTagger.h.

Referenced by discriminator(), and initialize().

CombinedSVSoftLeptonComputer CharmTagger::sl_computer_

private

Definition at line 40 of file CharmTagger.h.

Referenced by discriminator().

bool CharmTagger::use_adaBoost_

private

Definition at line 48 of file CharmTagger.h.

Referenced by initialize().

bool CharmTagger::use_condDB_

private

Definition at line 44 of file CharmTagger.h.

Referenced by initialize().

bool CharmTagger::use_GBRForest_

private

Definition at line 47 of file CharmTagger.h.

Referenced by initialize().

std::vector<MVAVar> CharmTagger::variables_

private

Definition at line 41 of file CharmTagger.h.

Referenced by CharmTagger(), discriminator(), and initialize().

edm::FileInPath CharmTagger::weight_file_

private

Definition at line 46 of file CharmTagger.h.

Referenced by initialize().