MTDTrackQualityMVA Class Reference

#include <MTDTrackQualityMVA.h>

Public Types
enum	VarID {   VarID::pt, VarID::eta, VarID::phi, VarID::chi2,   VarID::ndof, VarID::numberOfValidHits, VarID::numberOfValidPixelBarrelHits, VarID::numberOfValidPixelEndcapHits,   VarID::btlMatchChi2, VarID::btlMatchTimeChi2, VarID::etlMatchChi2, VarID::etlMatchTimeChi2,   VarID::mtdt, VarID::path_len }

Public Member Functions
	MTDTrackQualityMVA (std::string weights_file)
float	operator() (const reco::TrackRef &trk, const edm::ValueMap< int > &npixBarrels, const edm::ValueMap< int > &npixEndcaps, const edm::ValueMap< float > &btl_chi2s, const edm::ValueMap< float > &btl_time_chi2s, const edm::ValueMap< float > &etl_chi2s, const edm::ValueMap< float > &etl_time_chi2s, const edm::ValueMap< float > &tmtds, const edm::ValueMap< float > &trk_lengths) const

Private Attributes
std::unique_ptr< TMVAEvaluator >	mva_
std::vector< std::string >	spec_vars_
std::vector< std::string >	vars_

Detailed Description

Definition at line 31 of file MTDTrackQualityMVA.h.

Member Enumeration Documentation

enum MTDTrackQualityMVA::VarID

strong

Enumerator
pt
eta
phi
chi2
ndof
numberOfValidHits
numberOfValidPixelBarrelHits
numberOfValidPixelEndcapHits
btlMatchChi2
btlMatchTimeChi2
etlMatchChi2
etlMatchTimeChi2
mtdt
path_len

Definition at line 36 of file MTDTrackQualityMVA.h.

{ MTDTRACKQUALITYMVA_VARS(MTDBDTVAR_ENUM) };

Constructor & Destructor Documentation

MTDTrackQualityMVA::MTDTrackQualityMVA

(

std::string

weights_file

)

Definition at line 3 of file MTDTrackQualityMVA.cc.

References MTDBDTVAR_STRING, MTDTRACKQUALITYMVA_VARS, mva_, AlCaHLTBitMon_ParallelJobs::options, spec_vars_, AlCaHLTBitMon_QueryRunRegistry::string, and vars_.

                                                             {
  std::string options("!Color:Silent");
  std::string method("BDT");

  std::string vars_array[] = {MTDTRACKQUALITYMVA_VARS(MTDBDTVAR_STRING)};
  int nvars = sizeof(vars_array) / sizeof(vars_array[0]);
  vars_.assign(vars_array, vars_array + nvars);

  mva_ = std::make_unique<TMVAEvaluator>();
  mva_->initialize(options, method, weights_file, vars_, spec_vars_, true, false);  //use GBR, GradBoost
}

Member Function Documentation

float MTDTrackQualityMVA::operator()	(	const reco::TrackRef &	trk,
		const edm::ValueMap< int > &	npixBarrels,
		const edm::ValueMap< int > &	npixEndcaps,
		const edm::ValueMap< float > &	btl_chi2s,
		const edm::ValueMap< float > &	btl_time_chi2s,
		const edm::ValueMap< float > &	etl_chi2s,
		const edm::ValueMap< float > &	etl_time_chi2s,
		const edm::ValueMap< float > &	tmtds,
		const edm::ValueMap< float > &	trk_lengths
	)		const

Definition at line 15 of file MTDTrackQualityMVA.cc.

References btlMatchChi2, btlMatchTimeChi2, chi2, edm::ValueMap< T >::contains(), eta, etlMatchChi2, etlMatchTimeChi2, mtdt, mva_, ndof, numberOfValidHits, numberOfValidPixelBarrelHits, numberOfValidPixelEndcapHits, path_len, phi, pt, mathSSE::sqrt(), and vars_.

                                                                                  {
  std::map<std::string, float> vars;

  //---training performed only above 0.5 GeV
  constexpr float minPtForMVA = 0.5;
  if (trk->pt() < minPtForMVA)
    return -1;

  //---training performed only for tracks with MTD hits
  if (tmtds[trk] > 0) {
    vars.emplace(vars_[int(VarID::pt)], trk->pt());
    vars.emplace(vars_[int(VarID::eta)], trk->eta());
    vars.emplace(vars_[int(VarID::phi)], trk->phi());
    vars.emplace(vars_[int(VarID::chi2)], trk->chi2());
    vars.emplace(vars_[int(VarID::ndof)], trk->ndof());
    vars.emplace(vars_[int(VarID::numberOfValidHits)], trk->numberOfValidHits());
    vars.emplace(vars_[int(VarID::numberOfValidPixelBarrelHits)], npixBarrels[trk]);
    vars.emplace(vars_[int(VarID::numberOfValidPixelEndcapHits)], npixEndcaps[trk]);
    vars.emplace(vars_[int(VarID::btlMatchChi2)], btl_chi2s.contains(trk.id()) ? btl_chi2s[trk] : -1);
    vars.emplace(vars_[int(VarID::btlMatchTimeChi2)], btl_time_chi2s.contains(trk.id()) ? btl_time_chi2s[trk] : -1);
    vars.emplace(vars_[int(VarID::etlMatchChi2)], etl_chi2s.contains(trk.id()) ? etl_chi2s[trk] : -1);
    vars.emplace(vars_[int(VarID::etlMatchTimeChi2)], etl_time_chi2s.contains(trk.id()) ? etl_time_chi2s[trk] : -1);
    vars.emplace(vars_[int(VarID::mtdt)], tmtds[trk]);
    vars.emplace(vars_[int(VarID::path_len)], trk_lengths[trk]);
    return 1. / (1 + sqrt(2 / (1 + mva_->evaluate(vars, false)) - 1));  //return values between 0-1 (probability)
  } else
    return -1;
}

Member Data Documentation

std::unique_ptr<TMVAEvaluator> MTDTrackQualityMVA::mva_

private

Definition at line 52 of file MTDTrackQualityMVA.h.

Referenced by MTDTrackQualityMVA(), and operator()().

std::vector<std::string> MTDTrackQualityMVA::spec_vars_

private

Definition at line 51 of file MTDTrackQualityMVA.h.

Referenced by MTDTrackQualityMVA().

std::vector<std::string> MTDTrackQualityMVA::vars_

private

Definition at line 51 of file MTDTrackQualityMVA.h.

Referenced by MTDTrackQualityMVA(), and operator()().