#include <MVAValueMapProducer.h>

Inheritance diagram for MVAValueMapProducer< ParticleType >:

Public Member Functions
	MVAValueMapProducer (const edm::ParameterSet &)

	~MVAValueMapProducer () 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)

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
std::vector< std::string >	mvaCategoriesMapNames_

std::vector< std::unique_ptr< AnyMVAEstimatorRun2Base > >	mvaEstimators_

std::vector< std::string >	mvaRawValueMapNames_

std::vector< std::string >	mvaValueMapNames_

MultiTokenT< edm::View< ParticleType > >	src_

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<class ParticleType>
class MVAValueMapProducer< ParticleType >

Definition at line 25 of file MVAValueMapProducer.h.

Constructor & Destructor Documentation

template<class ParticleType >

MVAValueMapProducer< ParticleType >::MVAValueMapProducer ( const edm::ParameterSet & iConfig )

Definition at line 52 of file MVAValueMapProducer.h.

References beamerCreator::create(), Exception, reco::get(), edm::ParameterSet::getParameterSetVector(), MVAValueMapProducer< ParticleType >::mvaCategoriesMapNames_, MVAValueMapProducer< ParticleType >::mvaEstimators_, MVAValueMapProducer< ParticleType >::mvaRawValueMapNames_, MVAValueMapProducer< ParticleType >::mvaValueMapNames_, and AlCaHLTBitMon_QueryRunRegistry::string.

   : src_(consumesCollector(), iConfig, "src", "srcMiniAOD")
 
 {
   // Loop over the list of MVA configurations passed here from python and
   // construct all requested MVA estimators.
   const std::vector<edm::ParameterSet>& mvaEstimatorConfigs
     = iConfig.getParameterSetVector("mvaConfigurations");
 
   for( auto &imva : mvaEstimatorConfigs ){
 
     // The factory below constructs the MVA of the appropriate type based
     // on the "mvaName" which is the name of the derived MVA class (plugin)
     if( !imva.empty() ) {
 
       mvaEstimators_.emplace_back(AnyMVAEstimatorRun2Factory::get()->create(
                   imva.getParameter<std::string>("mvaName"), imva));
 
     } else
       throw cms::Exception(" MVA configuration not found: ")
         << " failed to find proper configuration for one of the MVAs in the main python script " << std::endl;
 
     mvaEstimators_.back()->setConsumes( consumesCollector() );
 
     //
     // Compose and save the names of the value maps to be produced
     //
 
     const std::string fullName = ( mvaEstimators_.back()->getName() +
                                    mvaEstimators_.back()->getTag()  );
 
     const std::string thisValueMapName      = fullName + "Values";
     const std::string thisRawValueMapName   = fullName + "RawValues";
     const std::string thisCategoriesMapName = fullName + "Categories";
 
     mvaValueMapNames_     .push_back( thisValueMapName      );
     mvaRawValueMapNames_  .push_back( thisRawValueMapName   );
     mvaCategoriesMapNames_.push_back( thisCategoriesMapName );
 
     // Declare the maps to the framework
     produces<edm::ValueMap<float>>(thisValueMapName     );
     produces<edm::ValueMap<float>>(thisRawValueMapName  );
     produces<edm::ValueMap<int>>  (thisCategoriesMapName);
 
   }
 
 }

template<class ParticleType >

MVAValueMapProducer< ParticleType >::~MVAValueMapProducer ( )

override

Definition at line 101 of file MVAValueMapProducer.h.

101 {

102 }

Member Function Documentation

template<class ParticleType >

void MVAValueMapProducer< ParticleType >::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 135 of file MVAValueMapProducer.h.

References edm::ConfigurationDescriptions::addDefault(), and edm::ParameterSetDescription::setUnknown().

                                                                                                      {
   //The following says we do not know what parameters are allowed so do no validation
   // Please change this to state exactly what you do use, even if it is no parameters
   edm::ParameterSetDescription desc;
   desc.setUnknown();
   descriptions.addDefault(desc);
 }

template<class ParticleType >

void MVAValueMapProducer< ParticleType >::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 105 of file MVAValueMapProducer.h.

References eostools::cat(), JetChargeProducer_cfi::exp, MultiTokenT< T >::getValidHandle(), mps_fire::i, MVAValueMapProducer< ParticleType >::mvaCategoriesMapNames_, MVAValueMapProducer< ParticleType >::mvaEstimators_, MVAValueMapProducer< ParticleType >::mvaRawValueMapNames_, MVAValueMapProducer< ParticleType >::mvaValueMapNames_, TrackRefitter_38T_cff::src, MVAValueMapProducer< ParticleType >::src_, and writeValueMap().

                                                                                              {
 
   edm::Handle<edm::View<ParticleType> > src = src_.getValidHandle(iEvent);
 
   // Loop over MVA estimators
   for( unsigned iEstimator = 0; iEstimator < mvaEstimators_.size(); iEstimator++ ){
 
     std::vector<float> mvaValues;
     std::vector<float> mvaRawValues;
     std::vector<int>   mvaCategories;
 
     // Loop over particles
     for (size_t i = 0; i < src->size(); ++i){
       auto iCand = src->ptrAt(i);
       int cat = -1; // Passed by reference to the mvaValue function to store the category
       const float response = mvaEstimators_[iEstimator]->mvaValue( iCand, iEvent, cat );
       mvaRawValues.push_back( response ); // The MVA score
       mvaValues.push_back( 2.0/(1.0+exp(-2.0*response))-1 ); // MVA output between -1 and 1
       mvaCategories.push_back( cat );
     } // end loop over particles
 
     writeValueMap(iEvent, src, mvaValues    , mvaValueMapNames_     [iEstimator] );
     writeValueMap(iEvent, src, mvaRawValues , mvaRawValueMapNames_  [iEstimator] );
     writeValueMap(iEvent, src, mvaCategories, mvaCategoriesMapNames_[iEstimator] );
 
   } // end loop over estimators
 
 }