CMS 3D CMS Logo

Namespaces | Functions | Variables
MLPFModel.h File Reference
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockElement.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"

Go to the source code of this file.

Namespaces

 reco
 fixed size matrix
 
 reco::mlpf
 

Functions

int reco::mlpf::argMax (std::vector< float > const &vec)
 
std::array< float, NUM_ELEMENT_FEATURES > reco::mlpf::getElementProperties (const reco::PFBlockElement &orig)
 
const std::vector< const reco::PFBlockElement * > reco::mlpf::getPFElements (const reco::PFBlockCollection &blocks)
 
reco::PFCandidate reco::mlpf::makeCandidate (int pred_pid, int pred_charge, float pred_e, float pred_eta, float pred_phi)
 
float reco::mlpf::normalize (float in)
 
void reco::mlpf::setCandidateRefs (reco::PFCandidate &cand, const std::vector< const reco::PFBlockElement * > elems, size_t ielem_originator)
 

Variables

static constexpr int reco::mlpf::BATCH_SIZE = 1
 
static const std::map< int, int > reco::mlpf::elem_type_encoding
 
static constexpr unsigned int reco::mlpf::IDX_CHARGE = 11
 
static constexpr unsigned int reco::mlpf::IDX_ENERGY = 10
 
static constexpr unsigned int reco::mlpf::IDX_ETA = 8
 
static constexpr unsigned int reco::mlpf::IDX_PHI = 9
 
static constexpr int reco::mlpf::LSH_BIN_SIZE = 100
 
static constexpr unsigned int reco::mlpf::NUM_CLASS = 7
 
static constexpr unsigned int reco::mlpf::NUM_ELEMENT_FEATURES = 15
 
static constexpr int reco::mlpf::NUM_MAX_ELEMENTS_BATCH = 20000
 
static constexpr unsigned int reco::mlpf::NUM_OUTPUTS = 12
 
static const std::vector< int > reco::mlpf::pdgid_encoding = {0, 1, 2, 11, 13, 22, 130, 211}