reco::tau::RecoTauMVAHelper Class Reference

#include <RecoTauMVAHelper.h>

Public Member Functions
const PhysicsTools::Variable::ValueList &	discriminants (const PFTauRef &tau) const
double	operator() (const PFTauRef &tau) const
	RecoTauMVAHelper (const std::string &name, const std::string eslabel="")
void	setEvent (const edm::Event &evt, const edm::EventSetup &es)
void	train (const PFTauRef &tau, bool target, double weight=1.0) const
	~RecoTauMVAHelper ()

Private Types
typedef boost::ptr_map < PhysicsTools::AtomicId, RecoTauDiscriminantPlugin >	PluginMap

Private Member Functions
void	fillValues (const reco::PFTauRef &tau) const
void	loadDiscriminantPlugins (const PhysicsTools::Calibration::MVAComputer &computer)

Private Attributes
PhysicsTools::MVAComputerCache	computer_
std::string	eslabel_
std::string	name_
PluginMap	plugins_
PhysicsTools::Variable::ValueList	values_

Detailed Description

Definition at line 37 of file RecoTauMVAHelper.h.

Member Typedef Documentation

typedef boost::ptr_map<PhysicsTools::AtomicId, RecoTauDiscriminantPlugin> reco::tau::RecoTauMVAHelper::PluginMap

private

Definition at line 63 of file RecoTauMVAHelper.h.

Constructor & Destructor Documentation

reco::tau::RecoTauMVAHelper::RecoTauMVAHelper ( const std::string &  name, const std::string  eslabel = ""  )

explicit

Definition at line 15 of file RecoTauMVAHelper.cc.

                                                           :
    name_(name), eslabel_(eslabel) {}

reco::tau::RecoTauMVAHelper::~RecoTauMVAHelper ( )

inline

Definition at line 41 of file RecoTauMVAHelper.h.

{}

Member Function Documentation

const PhysicsTools::Variable::ValueList & reco::tau::RecoTauMVAHelper::discriminants

(

const PFTauRef &

tau

)

const

Definition at line 87 of file RecoTauMVAHelper.cc.

References PhysicsTools::Variable::ValueList::clear(), fillValues(), and values_.

                                                         {
  values_.clear();
  fillValues(tau);
  return values_;
}

void reco::tau::RecoTauMVAHelper::fillValues ( const reco::PFTauRef &  tau ) const

private

Definition at line 63 of file RecoTauMVAHelper.cc.

References PhysicsTools::Variable::ValueList::add(), edm::detail::isnan(), python.pfTaus_cff::plugin, plugins_, metsig::tau, and values_.

Referenced by discriminants(), operator()(), and train().

                                                               {
  // Loop over the relevant discriminators and the output
  for (PluginMap::const_iterator plugin = plugins_.begin();
       plugin != plugins_.end(); ++plugin) {
    PhysicsTools::AtomicId id = plugin->first;
    std::vector<double> pluginOutput = (plugin->second)->operator()(tau);
    // Check for nans
    for(size_t instance = 0; instance < pluginOutput.size(); ++instance) {
      if (std::isnan(pluginOutput[instance])) {
        edm::LogError("CorruptedMVAInput") << "A nan was detected in"
            << " the tau MVA variable " << id << " returning zero instead!";
        pluginOutput[instance] = 0.0;
      }
    }
    //std::cout << "id: " << id << " first: " << pluginOutput[0] << std::endl;
    // Build values and copy into values vector
    std::for_each(pluginOutput.begin(), pluginOutput.end(),
                  boost::bind(&PhysicsTools::Variable::ValueList::add,
                              boost::ref(values_), id, _1));
  }
}

void reco::tau::RecoTauMVAHelper::loadDiscriminantPlugins ( const PhysicsTools::Calibration::MVAComputer &  computer )

private

Definition at line 41 of file RecoTauMVAHelper.cc.

References edm::ParameterSet::addParameter(), SurfaceDeformationFactory::create(), reco::tau::discPluginName(), reco::get(), PhysicsTools::Calibration::MVAComputer::inputSet, and plugins_.

Referenced by setEvent().

                                                    {
  typedef std::vector<PhysicsTools::Calibration::Variable> VarList;
  // List of input variables for this MVA.
  const VarList &vars = comp.inputSet;
  // Load the plugin for each of the Var if needed
  BOOST_FOREACH(const VarList::value_type& var, vars) {
    // Check to make sure it isn't a magic variable
    if (std::strncmp(var.name.c_str(), "__", 2) != 0) {
      // If we haven't added yet, build it.
      PhysicsTools::AtomicId varId(var.name);
      if (!plugins_.count(varId)) {
        edm::ParameterSet fakePSet;
        fakePSet.addParameter("name", "MVA_" + var.name);
        plugins_.insert(
            varId, RecoTauDiscriminantPluginFactory::get()->create(
                reco::tau::discPluginName(var.name), fakePSet));
      }
    }
  }
}

double reco::tau::RecoTauMVAHelper::operator()

(

const PFTauRef &

tau

)

const

Definition at line 94 of file RecoTauMVAHelper.cc.

References PhysicsTools::Variable::ValueList::clear(), computer_, PhysicsTools::MVAComputer::eval(), fillValues(), and values_.

                                                                 {
  // Clear output
  values_.clear();
  // Build the values
  fillValues(tau);
  // Call the MVA
  return computer_->eval(values_);
}

void reco::tau::RecoTauMVAHelper::setEvent	(	const edm::Event &	evt,
		const edm::EventSetup &	es
	)

Definition at line 19 of file RecoTauMVAHelper.cc.

References computer_, eslabel_, edm::EventSetup::get(), PhysicsTools::MVAComputerCache::get(), patZpeak::handle, loadDiscriminantPlugins(), name_, python.pfTaus_cff::plugin, plugins_, edm::ESHandle< class >::product(), and PhysicsTools::MVAComputerCache::update().

Referenced by RecoTauMVATrainer::analyze().

                                                         {
  // Update the event info for all of our discriminators
  BOOST_FOREACH(PluginMap::value_type plugin, plugins_) {
    plugin.second->setup(evt, es);
  }
  // Update our MVA from the DB
  using PhysicsTools::Calibration::MVAComputerContainer;
  edm::ESHandle<MVAComputerContainer> handle;
  if (eslabel_.size()) {
    es.get<TauTagMVAComputerRcd>().get(eslabel_, handle);
  } else {
    es.get<TauTagMVAComputerRcd>().get(handle);
  }
  const MVAComputerContainer *container = handle.product();
  // Load our MVA
  bool reload = computer_.update(container, name_.c_str());
  // If the MVA changed, update our list of discriminant plugins
  if (reload && computer_.get())
    loadDiscriminantPlugins(container->find(name_));
}

void reco::tau::RecoTauMVAHelper::train	(	const PFTauRef &	tau,
		bool	target,
		double	weight = 1.0
	)		const

Definition at line 103 of file RecoTauMVAHelper.cc.

References PhysicsTools::Variable::ValueList::add(), PhysicsTools::Variable::ValueList::clear(), computer_, PhysicsTools::MVAComputer::eval(), fillValues(), and values_.

                                                  {
  static const PhysicsTools::AtomicId kTargetId("__TARGET__");
  static const PhysicsTools::AtomicId kWeightId("__WEIGHT__");
  if (!computer_)
    return;
  values_.clear();
  values_.add(kTargetId, target);
  values_.add(kWeightId, weight);
  // Build the discriminant values
  fillValues(tau);
  computer_->eval(values_);
}

Member Data Documentation

PhysicsTools::MVAComputerCache reco::tau::RecoTauMVAHelper::computer_

private

Definition at line 60 of file RecoTauMVAHelper.h.

Referenced by operator()(), setEvent(), and train().

std::string reco::tau::RecoTauMVAHelper::eslabel_

private

Definition at line 59 of file RecoTauMVAHelper.h.

Referenced by setEvent().

std::string reco::tau::RecoTauMVAHelper::name_

private

Definition at line 57 of file RecoTauMVAHelper.h.

Referenced by Config.Process::fillProcessDesc(), and setEvent().

PluginMap reco::tau::RecoTauMVAHelper::plugins_

private

Definition at line 64 of file RecoTauMVAHelper.h.

Referenced by fillValues(), loadDiscriminantPlugins(), and setEvent().

PhysicsTools::Variable::ValueList reco::tau::RecoTauMVAHelper::values_

mutableprivate

Definition at line 70 of file RecoTauMVAHelper.h.

Referenced by discriminants(), fillValues(), operator()(), and train().