l1tpf::HGC3DClusterEgID Class Reference

#include <HGC3DClusterEgID.h>

Classes
class	Var

Public Member Functions
	HGC3DClusterEgID (const edm::ParameterSet &pset)
std::string	method ()
float	passID (l1t::HGCalMulticluster c, l1t::PFCluster &cpf)
void	prepareTMVA ()

Private Attributes
bool	isPUFilter_
std::string	method_
StringCutObjectSelector < l1t::HGCalMulticluster >	preselection_
std::unique_ptr< TMVA::Reader >	reader_
std::vector< Var >	variables_
std::string	weightsFile_
StringObjectFunction < l1t::HGCalMulticluster >	wp_

Detailed Description

Definition at line 21 of file HGC3DClusterEgID.h.

Constructor & Destructor Documentation

l1tpf::HGC3DClusterEgID::HGC3DClusterEgID

(

const edm::ParameterSet &

pset

)

Definition at line 4 of file HGC3DClusterEgID.cc.

References edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, and variables_.

    : isPUFilter_(pset.getParameter<bool>("isPUFilter")),
      preselection_(pset.getParameter<std::string>("preselection")),
      method_(pset.getParameter<std::string>("method")),
      weightsFile_(pset.getParameter<std::string>("weightsFile")),
      reader_(new TMVA::Reader()),
      wp_(pset.getParameter<std::string>("wp")) {
  // first create all the variables
  for (const auto &psvar : pset.getParameter<std::vector<edm::ParameterSet>>("variables")) {
    variables_.emplace_back(psvar.getParameter<std::string>("name"), psvar.getParameter<std::string>("value"));
  }
}

Member Function Documentation

std::string l1tpf::HGC3DClusterEgID::method ( )

inline

Definition at line 29 of file HGC3DClusterEgID.h.

References method_.

Referenced by l1tpf::PFClusterProducerFromHGC3DClusters::PFClusterProducerFromHGC3DClusters().

{ return method_; }

float l1tpf::HGC3DClusterEgID::passID	(	l1t::HGCalMulticluster	c,
		l1t::PFCluster &	cpf
	)

Definition at line 28 of file HGC3DClusterEgID.cc.

References l1t::PFCluster::setEgVsPionMVAOut(), l1t::PFCluster::setEgVsPUMVAOut(), and isotrackApplyRegressor::var.

                                                                             {
  if (preselection_(c)) {
    for (auto &var : variables_)
      var.fill(c);
    float mvaOut = reader_->EvaluateMVA(method_);
    if (isPUFilter_)
      cpf.setEgVsPUMVAOut(mvaOut);
    else
      cpf.setEgVsPionMVAOut(mvaOut);
    return (mvaOut > wp_(c) ? 1 : 0);
  } else {
    if (isPUFilter_)
      cpf.setEgVsPUMVAOut(-100.0);
    else
      cpf.setEgVsPionMVAOut(-100.0);
    return 0;
  }
}

void l1tpf::HGC3DClusterEgID::prepareTMVA

(

)

Definition at line 17 of file HGC3DClusterEgID.cc.

References edm::FileInPath::fullPath(), reco::details::loadTMVAWeights(), and isotrackApplyRegressor::var.

Referenced by l1tpf::PFClusterProducerFromHGC3DClusters::PFClusterProducerFromHGC3DClusters().

                                        {
  // Declare the variables
  for (auto &var : variables_)
    var.declare(*reader_);
  // then read the weights
  if (weightsFile_[0] != '/' && weightsFile_[0] != '.') {
    weightsFile_ = edm::FileInPath(weightsFile_).fullPath();
  }
  reco::details::loadTMVAWeights(&*reader_, method_, weightsFile_);
}

Member Data Documentation

bool l1tpf::HGC3DClusterEgID::isPUFilter_

private

Definition at line 44 of file HGC3DClusterEgID.h.

std::string l1tpf::HGC3DClusterEgID::method_

private

Definition at line 47 of file HGC3DClusterEgID.h.

Referenced by method().

StringCutObjectSelector<l1t::HGCalMulticluster> l1tpf::HGC3DClusterEgID::preselection_

private

Definition at line 45 of file HGC3DClusterEgID.h.

std::unique_ptr<TMVA::Reader> l1tpf::HGC3DClusterEgID::reader_

private

Definition at line 48 of file HGC3DClusterEgID.h.

std::vector<Var> l1tpf::HGC3DClusterEgID::variables_

private

Definition at line 46 of file HGC3DClusterEgID.h.

Referenced by HGC3DClusterEgID().

std::string l1tpf::HGC3DClusterEgID::weightsFile_

private

Definition at line 47 of file HGC3DClusterEgID.h.

StringObjectFunction<l1t::HGCalMulticluster> l1tpf::HGC3DClusterEgID::wp_

private

Definition at line 49 of file HGC3DClusterEgID.h.