CMS 3D CMS Logo

Classes | Namespaces | Functions
deep_helpers.h File Reference
#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1D.h"
#include "DataFormats/BTauReco/interface/TaggingVariable.h"
#include "TrackingTools/IPTools/interface/IPTools.h"
#include "DataFormats/BTauReco/interface/CandIPTagInfo.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Candidate/interface/VertexCompositePtrCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/PatCandidates/interface/Jet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "PhysicsTools/ONNXRuntime/interface/ONNXRuntime.h"
#include <iostream>
#include <fstream>
#include <algorithm>
#include <numeric>
#include <nlohmann/json.hpp>

Go to the source code of this file.

Classes

struct  btagbtvdeep::PreprocessParams
 
struct  btagbtvdeep::PreprocessParams::VarInfo
 

Namespaces

 btagbtvdeep
 

Functions

const float btagbtvdeep::catch_infs (const float in, const float replace_value=0.)
 
const float btagbtvdeep::catch_infs_and_bound (const float in, const float replace_value, const float lowerbound, const float upperbound, const float offset=0., const bool use_offsets=true)
 
int btagbtvdeep::center_norm_pad (const std::vector< float > &input, float center, float scale, unsigned min_length, unsigned max_length, std::vector< float > &datavec, int startval, float pad_value=0, float replace_inf_value=0, float min=0, float max=-1)
 
int btagbtvdeep::center_norm_pad_halfRagged (const std::vector< float > &input, float center, float scale, unsigned target_length, std::vector< float > &datavec, int startval, float pad_value=0, float replace_inf_value=0, float min=0, float max=-1)
 
template<typename T >
int btagbtvdeep::dump_vector (reco::TaggingVariableList &from, T *to, reco::btau::TaggingVariableName name, const size_t max)
 
std::pair< float, float > btagbtvdeep::getDRSubjetFeatures (const reco::Jet &jet, const reco::Candidate *cand)
 
float btagbtvdeep::lost_inner_hits_from_pfcand (const reco::PFCandidate &pfcand)
 
float btagbtvdeep::mindrsvpfcand (const std::vector< reco::VertexCompositePtrCandidate > &svs, const reco::Candidate *cand, float mindr=0.4)
 
void btagbtvdeep::ParticleNetConstructor (const edm::ParameterSet &Config_, bool doExtra, std::vector< std::string > &input_names_, std::unordered_map< std::string, PreprocessParams > &prep_info_map_, std::vector< std::vector< int64_t >> &input_shapes_, std::vector< unsigned > &input_sizes_, cms::Ort::FloatArrays *data_)
 
float btagbtvdeep::quality_from_pfcand (const reco::PFCandidate &pfcand)
 
template<typename SVType , typename PVType >
bool btagbtvdeep::sv_vertex_comparator (const SVType &sva, const SVType &svb, const PVType &pv)
 
Measurement1D btagbtvdeep::vertexD3d (const reco::VertexCompositePtrCandidate &svcand, const reco::Vertex &pv)
 
float btagbtvdeep::vertexDdotP (const reco::VertexCompositePtrCandidate &sv, const reco::Vertex &pv)
 
Measurement1D btagbtvdeep::vertexDxy (const reco::VertexCompositePtrCandidate &svcand, const reco::Vertex &pv)
 
float btagbtvdeep::vtx_ass_from_pfcand (const reco::PFCandidate &pfcand, int pv_ass_quality, const reco::VertexRef &pv)