BaseMVAValueMapProducer< T > Class Template Reference

#include <PhysicsTools/PatAlgos/plugins/BaseMVAValueMapProducer.cc>

Inheritance diagram for BaseMVAValueMapProducer< T >:

Public Member Functions
	BaseMVAValueMapProducer (const edm::ParameterSet &iConfig)
void	setValue (const std::string var, float val)
	~BaseMVAValueMapProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
static edm::ParameterSetDescription	getDescription ()

Private Member Functions
void	beginStream (edm::StreamID) override
void	endStream () override
virtual void	fillAdditionalVariables (const T &)
void	produce (edm::Event &, const edm::EventSetup &) override
virtual void	readAdditionalCollections (edm::Event &, const edm::EventSetup &)
	to be implemented in derived classes, filling values for additional variables More...

Private Attributes
std::vector< std::pair< std::string, StringObjectFunction< T, true > > >	funcs_
bool	isClassifier_
std::string	name_
std::map< std::string, size_t >	positions_
TMVA::Reader	reader_
edm::EDGetTokenT< edm::View< T > >	src_
std::vector< float >	values_
std::vector< std::string >	variablesOrder_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

template<typename T>
class BaseMVAValueMapProducer< T >

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 50 of file BaseMVAValueMapProducer.h.

Constructor & Destructor Documentation

template<typename T>

BaseMVAValueMapProducer< T >::BaseMVAValueMapProducer ( const edm::ParameterSet &  iConfig )

inlineexplicit

Definition at line 52 of file BaseMVAValueMapProducer.h.

                                                                  :
    src_(consumes<edm::View<T>>(iConfig.getParameter<edm::InputTag>("src"))),
    variablesOrder_(iConfig.getParameter<std::vector<std::string>>("variablesOrder")),
    name_(iConfig.getParameter<std::string>("name")),
    isClassifier_(iConfig.getParameter<bool>("isClassifier"))  
  {
      edm::ParameterSet const & varsPSet = iConfig.getParameter<edm::ParameterSet>("variables");
      for (const std::string & vname : varsPSet.getParameterNamesForType<std::string>()) {
      funcs_.emplace_back(std::pair<std::string,StringObjectFunction<T,true>>(vname,varsPSet.getParameter<std::string>(vname)));
      }

      values_.resize(variablesOrder_.size());
      size_t i=0;
      for(const auto & v : variablesOrder_){
    positions_[v]=i;
    reader_.AddVariable(v,(&values_.front())+i);
    i++;
      }
//      reader_.BookMVA(name_,iConfig.getParameter<edm::FileInPath>("weightFile").fullPath() );
      reco::details::loadTMVAWeights(&reader_, name_, iConfig.getParameter<edm::FileInPath>("weightFile").fullPath());
      produces<edm::ValueMap<float>>();

  }

template<typename T>

BaseMVAValueMapProducer< T >::~BaseMVAValueMapProducer ( )

inlineoverride

Definition at line 75 of file BaseMVAValueMapProducer.h.

{}

Member Function Documentation

template<typename T>

void BaseMVAValueMapProducer< T >::beginStream ( edm::StreamID  )

inlineoverrideprivate

Definition at line 85 of file BaseMVAValueMapProducer.h.

{};

template<typename T>

void BaseMVAValueMapProducer< T >::endStream ( )

inlineoverrideprivate

Definition at line 87 of file BaseMVAValueMapProducer.h.

{};

template<typename T>

virtual void BaseMVAValueMapProducer< T >::fillAdditionalVariables ( const T &  )

inlineprivatevirtual

Reimplemented in BJetEnergyRegressionMVA.

Definition at line 91 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< T >::produce().

{}

template<typename T >

void BaseMVAValueMapProducer< T >::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 147 of file BaseMVAValueMapProducer.h.

References edm::ConfigurationDescriptions::add(), BaseMVAValueMapProducer< T >::getDescription(), and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by BaseMVAValueMapProducer< pat::Jet >::setValue().

                                                                                       {
  edm::ParameterSetDescription desc = getDescription();
  std::string modname;
  if (typeid(T) == typeid(pat::Jet)) modname+="Jet";
  else if (typeid(T) == typeid(pat::Muon)) modname+="Muon";
  else if (typeid(T) == typeid(pat::Electron)) modname+="Ele";
  modname+="BaseMVAValueMapProducer";
  descriptions.add(modname,desc);
}

template<typename T >

edm::ParameterSetDescription BaseMVAValueMapProducer< T >::getDescription ( )

static

Definition at line 132 of file BaseMVAValueMapProducer.h.

References edm::ParameterSetDescription::add(), edm::ParameterSetDescription::setAllowAnything(), AlCaHLTBitMon_QueryRunRegistry::string, and objects.autophobj::variables.

Referenced by BaseMVAValueMapProducer< T >::fillDescriptions(), BJetEnergyRegressionMVA::fillDescriptions(), and BaseMVAValueMapProducer< pat::Jet >::setValue().

                                          {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("src")->setComment("input physics object collection");
  desc.add<std::vector<std::string>>("variablesOrder")->setComment("ordered list of MVA input variable names");
  desc.add<std::string>("name")->setComment("output score variable name");
  desc.add<bool>("isClassifier")->setComment("is a classifier discriminator");
  edm::ParameterSetDescription variables;
  variables.setAllowAnything();
  desc.add<edm::ParameterSetDescription>("variables", variables)->setComment("list of input variable definitions");
  desc.add<edm::FileInPath>("weightFile")->setComment("xml weight file");
  return desc;
}

template<typename T >

void BaseMVAValueMapProducer< T >::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 107 of file BaseMVAValueMapProducer.h.

References edm::helper::Filler< Map >::fill(), BaseMVAValueMapProducer< T >::fillAdditionalVariables(), objects.autophobj::filler, BaseMVAValueMapProducer< T >::funcs_, edm::Event::getByToken(), edm::helper::Filler< Map >::insert(), BaseMVAValueMapProducer< T >::isClassifier_, eostools::move(), BaseMVAValueMapProducer< T >::name_, connectstrParser::o, AlCaHLTBitMon_ParallelJobs::p, BaseMVAValueMapProducer< T >::positions_, edm::Event::put(), BaseMVAValueMapProducer< T >::readAdditionalCollections(), BaseMVAValueMapProducer< T >::reader_, TrackRefitter_38T_cff::src, BaseMVAValueMapProducer< T >::src_, and BaseMVAValueMapProducer< T >::values_.

Referenced by BaseMVAValueMapProducer< pat::Jet >::beginStream().

{
  edm::Handle<edm::View<T>> src;
  iEvent.getByToken(src_, src);
  readAdditionalCollections(iEvent,iSetup);
  
  std::vector<float> mvaOut;
  mvaOut.reserve(src->size());
  for(auto const & o: *src) {
    for(auto const & p : funcs_ ){
        values_[positions_[p.first]]=p.second(o);
    }
        fillAdditionalVariables(o);
    mvaOut.push_back(isClassifier_ ? reader_.EvaluateMVA(name_) : reader_.EvaluateRegression(name_)[0]);
  }
  std::unique_ptr<edm::ValueMap<float>> mvaV(new edm::ValueMap<float>());
  edm::ValueMap<float>::Filler filler(*mvaV);
  filler.insert(src,mvaOut.begin(),mvaOut.end());
  filler.fill();
  iEvent.put(std::move(mvaV));

}

template<typename T>

virtual void BaseMVAValueMapProducer< T >::readAdditionalCollections ( edm::Event &  , const edm::EventSetup &    )

inlineprivatevirtual

to be implemented in derived classes, filling values for additional variables

Reimplemented in BJetEnergyRegressionMVA.

Definition at line 90 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< T >::produce().

{}

template<typename T>

void BaseMVAValueMapProducer< T >::setValue ( const std::string  var, float  val  )

inline

Definition at line 77 of file BaseMVAValueMapProducer.h.

Referenced by BJetEnergyRegressionMVA::fillAdditionalVariables().

                                               {
    values_[positions_[var]]=val;
  }

Member Data Documentation

template<typename T>

std::vector<std::pair<std::string,StringObjectFunction<T,true> > > BaseMVAValueMapProducer< T >::funcs_

private

Definition at line 96 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), and BaseMVAValueMapProducer< T >::produce().

template<typename T>

bool BaseMVAValueMapProducer< T >::isClassifier_

private

Definition at line 101 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< T >::produce().

template<typename T>

std::string BaseMVAValueMapProducer< T >::name_

private

Definition at line 100 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), and BaseMVAValueMapProducer< T >::produce().

template<typename T>

std::map<std::string,size_t> BaseMVAValueMapProducer< T >::positions_

private

Definition at line 95 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), BaseMVAValueMapProducer< T >::produce(), and BaseMVAValueMapProducer< pat::Jet >::setValue().

template<typename T>

TMVA::Reader BaseMVAValueMapProducer< T >::reader_

private

Definition at line 99 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), and BaseMVAValueMapProducer< T >::produce().

template<typename T>

edm::EDGetTokenT<edm::View<T> > BaseMVAValueMapProducer< T >::src_

private

Definition at line 94 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< T >::produce().

template<typename T>

std::vector<float> BaseMVAValueMapProducer< T >::values_

private

Definition at line 98 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), BaseMVAValueMapProducer< T >::produce(), and BaseMVAValueMapProducer< pat::Jet >::setValue().

template<typename T>

std::vector<std::string> BaseMVAValueMapProducer< T >::variablesOrder_

private

Definition at line 97 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer().