pat::MuonMvaIDEstimator Class Reference

#include <MuonMvaIDEstimator.h>

Public Member Functions
std::vector< float >	computeMVAID (const pat::Muon &imuon) const
	MuonMvaIDEstimator (const edm::FileInPath &weightsfile)
	~MuonMvaIDEstimator ()=default

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)
static void	globalEndJob (const cms::Ort::ONNXRuntime *)

Private Attributes
std::vector< std::string >	flav_names_
std::vector< std::string >	input_names_
std::unique_ptr< const cms::Ort::ONNXRuntime >	randomForest_

Detailed Description

Definition at line 21 of file MuonMvaIDEstimator.h.

Constructor & Destructor Documentation

MuonMvaIDEstimator::MuonMvaIDEstimator

(

const edm::FileInPath &

weightsfile

)

Definition at line 13 of file MuonMvaIDEstimator.cc.

References edm::FileInPath::fullPath(), and LogDebug.

                                                                       {
  randomForest_ = std::make_unique<ONNXRuntime>(weightsfile.fullPath());
  LogDebug("MuonMvaIDEstimator") << randomForest_.get();
}

pat::MuonMvaIDEstimator::~MuonMvaIDEstimator ( )

default

Member Function Documentation

std::vector< float > MuonMvaIDEstimator::computeMVAID

(

const pat::Muon &

imuon

)

const

Definition at line 32 of file MuonMvaIDEstimator.cc.

References arbitrationType, cms::cuda::assert(), reco::MuonQuality::chi2LocalPosition, reco::Muon::combinedQuality(), PVValHelper::eta, reco::LeafCandidate::eta(), pat::Muon::globalTrack(), pat::Muon::innerTrack(), reco::Muon::isGlobalMuon(), edm::Ref< C, T, F >::isNonnull(), LogDebug, reco::Muon::numberOfMatchedStations(), DiDispStaMuonMonitor_cfi::pt, reco::LeafCandidate::pt(), pat::Muon::segmentCompatibility(), and reco::MuonQuality::trkKink.

                                                                           {
  const float local_chi2 = muon.combinedQuality().chi2LocalPosition;
  const float kink = muon.combinedQuality().trkKink;
  const float segment_comp = muon.segmentCompatibility(arbitrationType);
  const float n_MatchedStations = muon.numberOfMatchedStations();
  const float pt = muon.pt();
  const float eta = muon.eta();
  const float global_muon = muon.isGlobalMuon();
  float Valid_pixel;
  float tracker_layers;
  float validFraction;
  if (muon.innerTrack().isNonnull()) {
    Valid_pixel = muon.innerTrack()->hitPattern().numberOfValidPixelHits();
    tracker_layers = muon.innerTrack()->hitPattern().trackerLayersWithMeasurement();
    validFraction = muon.innerTrack()->validFraction();
  } else {
    Valid_pixel = -99.;
    tracker_layers = -99.0;
    validFraction = -99.0;
  }
  float norm_chi2;
  float n_Valid_hits;
  if (muon.globalTrack().isNonnull()) {
    norm_chi2 = muon.globalTrack()->normalizedChi2();
    n_Valid_hits = muon.globalTrack()->hitPattern().numberOfValidMuonHits();
  } else if (muon.innerTrack().isNonnull()) {
    norm_chi2 = muon.innerTrack()->normalizedChi2();
    n_Valid_hits = muon.innerTrack()->hitPattern().numberOfValidMuonHits();
  } else {
    norm_chi2 = -99;
    n_Valid_hits = -99;
  }
  const std::vector<std::string> input_names_{"float_input"};
  std::vector<float> vars = {global_muon,
                             validFraction,
                             norm_chi2,
                             local_chi2,
                             kink,
                             segment_comp,
                             n_Valid_hits,
                             n_MatchedStations,
                             Valid_pixel,
                             tracker_layers,
                             pt,
                             eta};
  const std::vector<std::string> flav_names_{"probBAD", "probGOOD"};
  cms::Ort::FloatArrays input_values_;
  input_values_.emplace_back(vars);
  std::vector<float> outputs;
  LogDebug("MuonMvaIDEstimator") << randomForest_.get();
  outputs = randomForest_->run(input_names_, input_values_, {}, {"probabilities"})[0];
  assert(outputs.size() == flav_names_.size());
  return outputs;
}

void MuonMvaIDEstimator::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 18 of file MuonMvaIDEstimator.cc.

References edm::ParameterSetDescription::add(), edm::ConfigurationDescriptions::addWithDefaultLabel(), and submitPVResolutionJobs::desc.

                                                                                    {
  edm::ParameterSetDescription desc;
  desc.add<edm::FileInPath>("mvaIDTrainingFile", edm::FileInPath("RecoMuon/MuonIdentification/data/mvaID.onnx"));
  desc.add<std::vector<std::string>>("flav_names",
                                     std::vector<std::string>{
                                         "probBAD",
                                         "probGOOD",
                                     });

  descriptions.addWithDefaultLabel(desc);
}

void MuonMvaIDEstimator::globalEndJob ( const cms::Ort::ONNXRuntime *  cache )

static

Definition at line 30 of file MuonMvaIDEstimator.cc.

{}

Member Data Documentation

std::vector<std::string> pat::MuonMvaIDEstimator::flav_names_

private

Definition at line 31 of file MuonMvaIDEstimator.h.

std::vector<std::string> pat::MuonMvaIDEstimator::input_names_

private

Definition at line 32 of file MuonMvaIDEstimator.h.

std::unique_ptr<const cms::Ort::ONNXRuntime> pat::MuonMvaIDEstimator::randomForest_

private

Definition at line 33 of file MuonMvaIDEstimator.h.