CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
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::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_pt, float pred_eta, float pred_sin_phi, float pred_cos_phi, float pred_e)
 
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 = 8
 
static constexpr unsigned int reco::mlpf::IDX_CLASS = 7
 
static constexpr unsigned int reco::mlpf::IDX_COS_PHI = 12
 
static constexpr unsigned int reco::mlpf::IDX_ENERGY = 13
 
static constexpr unsigned int reco::mlpf::IDX_ETA = 10
 
static constexpr unsigned int reco::mlpf::IDX_PT = 9
 
static constexpr unsigned int reco::mlpf::IDX_SIN_PHI = 11
 
static constexpr int reco::mlpf::LSH_BIN_SIZE = 64
 
static constexpr unsigned int reco::mlpf::NUM_ELEMENT_FEATURES = 25
 
static constexpr int reco::mlpf::NUM_MAX_ELEMENTS_BATCH = 200 * LSH_BIN_SIZE
 
static constexpr unsigned int reco::mlpf::NUM_OUTPUT_FEATURES = 14
 
static const std::vector< int > reco::mlpf::pdgid_encoding = {0, 211, 130, 1, 2, 22, 11, 13}
 
static constexpr float reco::mlpf::PI_MASS = 0.13957