SelectorMVA Class Reference

MVA based selector for tagging curl tracks in Belle and Belle II. More...

#include <SelectorMVA.h>

Inheritance diagram for SelectorMVA:

Public Member Functions
	SelectorMVA (bool belleFlag, bool trainFlag, std::string tFileName)
	Constructor.
	~SelectorMVA ()
	Destructor.
virtual float	getResponse (Particle *iPart, Particle *jPart) override
	Selector response that this pair of particles come from the same mc/actual particle.
virtual float	getOptimalResponseCut () override
	returns optimal cut to use with selector
virtual std::vector< float >	getVariables (Particle *iPart, Particle *jPart) override
	returns vector of variables used by this selector.
virtual void	initialize () override
	initialize whatever needs to be initialized (root file etc)
virtual void	collectTrainingInfo (Particle *iPart, Particle *jPart) override
	collect training data and save to a root file
virtual void	finalize () override
	finalize whatever needs to be finalized (train the MVA)
void	initializeMVA ()
	initialize the MVA Expert

Private Member Functions
void	updateVariables (Particle *iPart, Particle *jPart)
	updates the value of the MVA variable

Private Attributes
bool	m_TrainFlag
	applying mva or training it
std::string	m_TFileName
	name of output file for training data
TFile *	m_TFile
	output file for training data
TTree *	m_TTree
	training data tree
std::unique_ptr< DBObjPtr< DatabaseRepresentationOfWeightfile > >	m_weightfile_representation
	Database pointer to the Database representation of the weightfile.
MVA::Weightfile	m_weightfile
	mva weightfile
MVA::GeneralOptions	m_generalOptions
	mva general options (for the expert)
MVA::FastBDTExpert	m_expert
	mva expert
std::string	m_identifier
	mva identifier
std::vector< std::string >	m_datafiles
	name of datafile
std::vector< std::string >	m_variables
	names of variables used by mva
std::string	m_target_variable
	name of target variable (isCurl)
Float_t	m_PPhi
	angle between particle momentum vectors
Float_t	m_ChargeProduct
	charge(p1) * charge(p2)
Float_t	m_PtDiffEW
	error weighted particle Pt difference
Float_t	m_PzDiffEW
	error weighted particle Pz difference
Float_t	m_TrackD0DiffEW
	error weighted track D0 difference
Float_t	m_TrackZ0DiffEW
	error weighted track Z0 difference
Float_t	m_TrackTanLambdaDiffEW
	error weighted track tan lambda diff difference
Float_t	m_TrackPhi0DiffEW
	error weighted track Phi0 difference
Float_t	m_TrackOmegaDiffEW
	error weighted track Omega difference
Bool_t	m_IsCurl
	isCurl Truth

Detailed Description

MVA based selector for tagging curl tracks in Belle and Belle II.

Definition at line 34 of file SelectorMVA.h.

Constructor & Destructor Documentation

SelectorMVA()

SelectorMVA	(	bool	belleFlag,
		bool	trainFlag,
		std::string	tFileName
	)

Constructor.

Definition at line 20 of file SelectorMVA.cc.

{
  m_TrainFlag = trainFlag;
  m_TFileName = tFileName;
  if (belleFlag) {
    m_identifier = "CurlTagger_FastBDT_Belle";
  } else {
    m_identifier = "CurlTagger_FastBDT_BelleII";
  }
}

Member Function Documentation

collectTrainingInfo()

void collectTrainingInfo ( Particle *  iPart, Particle *  jPart  )

overridevirtual

collect training data and save to a root file

Reimplemented from Selector.

Definition at line 73 of file SelectorMVA.cc.

{
  updateVariables(iPart, jPart);
  m_TTree -> Fill();
}

finalize()

void finalize ( )

overridevirtual

finalize whatever needs to be finalized (train the MVA)

Reimplemented from Selector.

Definition at line 135 of file SelectorMVA.cc.

{
  if (m_TrainFlag) {
    m_TFile -> cd();
    m_TTree -> Write();
    m_TFile -> Write();
    m_TFile -> Close();
  }
}

getOptimalResponseCut()

float getOptimalResponseCut ( )

overridevirtual

returns optimal cut to use with selector

Reimplemented from Selector.

Definition at line 122 of file SelectorMVA.cc.

{
  if (m_TrainFlag) {
    return 0.5;
  }
  std::string elementIdentfier = "optimal_cut";
  if (!m_weightfile.containsElement(elementIdentfier)) {
    B2FATAL("No optimal cut stored in curlTagger MVA payload!");
  }
  // require the default value for the compiler to deduce the template class
  return m_weightfile.getElement(elementIdentfier, 0.5);
}

getResponse()

float getResponse ( Particle *  iPart, Particle *  jPart  )

overridevirtual

Selector response that this pair of particles come from the same mc/actual particle.

Implements Selector.

Definition at line 145 of file SelectorMVA.cc.

{
  MVA::SingleDataset dataset(m_generalOptions, getVariables(iPart, jPart));
  return m_expert.apply(dataset)[0];
}

getVariables()

std::vector< float > getVariables ( Particle *  iPart, Particle *  jPart  )

overridevirtual

returns vector of variables used by this selector.

Implements Selector.

Definition at line 65 of file SelectorMVA.cc.

{
  updateVariables(iPart, jPart);
  return {m_PPhi, m_ChargeProduct, m_PtDiffEW,
          m_PzDiffEW, m_TrackD0DiffEW, m_TrackZ0DiffEW,
          m_TrackTanLambdaDiffEW, m_TrackPhi0DiffEW, m_TrackOmegaDiffEW};
}

initialize()

void initialize ( )

overridevirtual

initialize whatever needs to be initialized (root file etc)

Reimplemented from Selector.

Definition at line 79 of file SelectorMVA.cc.

{
  if (m_TrainFlag) {
    //make training data
    m_TFile = TFile::Open(m_TFileName.c_str(), "RECREATE");
    m_TTree = new TTree("ntuple", "Training Data for the Curl Tagger MVA");
 
    m_TTree -> Branch("PPhi", &m_PPhi, "PPhi/F");
    m_TTree -> Branch("ChargeProduct", &m_ChargeProduct, "ChargeProduct/F");
    m_TTree -> Branch("PtDiffEW", &m_PtDiffEW, "PtDiffEW/F");
    m_TTree -> Branch("PzDiffEW", &m_PzDiffEW, "PzDiffEW/F");
    m_TTree -> Branch("TrackD0DiffEW", &m_TrackD0DiffEW, "TrackD0DiffEW/F");
    m_TTree -> Branch("TrackZ0DiffEW", &m_TrackZ0DiffEW, "TrackZ0DiffEW/F");
    m_TTree -> Branch("TrackTanLambdaDiffEW", &m_TrackTanLambdaDiffEW, "TrackTanLambdaDiffEW/F");
    m_TTree -> Branch("TrackPhi0DiffEW", &m_TrackPhi0DiffEW, "TrackPhi0DiffEW/F");
    m_TTree -> Branch("TrackOmegaDiffEW", &m_TrackOmegaDiffEW, "TrackOmegaDiffEW/F");
 
    m_TTree -> Branch("IsCurl", &m_IsCurl, "IsCurl/O");
 
    m_target_variable = "IsCurl";
    m_variables = {"PPhi", "ChargeProduct", "PtDiffEW",
                   "PzDiffEW", "TrackD0DiffEW", "TrackZ0DiffEW",
                   "TrackTanLambdaDiffEW", "TrackPhi0DiffEW", "TrackOmegaDiffEW"
                  };
 
  } else {
    // normal application
    m_weightfile_representation = std::make_unique<DBObjPtr<DatabaseRepresentationOfWeightfile>>(
                                    MVA::makeSaveForDatabase(m_identifier));
    (*m_weightfile_representation.get()).addCallback([this]() { initializeMVA();});
    initializeMVA();
  }
}

initializeMVA()

void initializeMVA

(

)

initialize the MVA Expert

Definition at line 113 of file SelectorMVA.cc.

{
  std::stringstream ss((*m_weightfile_representation)->m_data);
  m_weightfile = MVA::Weightfile::loadFromStream(ss);
  m_weightfile.getOptions(m_generalOptions);
  m_expert.load(m_weightfile);
}

updateVariables()

void updateVariables ( Particle *  iPart, Particle *  jPart  )

private

updates the value of the MVA variable

Definition at line 33 of file SelectorMVA.cc.

{
  if (m_TrainFlag) {
    m_IsCurl = Variable::genParticleIndex(iPart) == Variable::genParticleIndex(jPart);
  }
  m_ChargeProduct = iPart->getCharge() * jPart->getCharge();
 
  m_PPhi  = acos(iPart->getMomentum().Unit().Dot(jPart->getMomentum().Unit()));
 
  m_PtDiffEW = std::abs(Variable::particlePt(iPart) - Variable::particlePt(jPart)) / sqrt(pow(Variable::particlePtErr(
                 iPart), 2) + pow(Variable::particlePtErr(jPart), 2));
 
  m_PzDiffEW = std::abs(Variable::particlePz(iPart) - Variable::particlePz(jPart)) / sqrt(pow(Variable::particlePzErr(
                 iPart), 2) + pow(Variable::particlePzErr(jPart), 2));
 
  m_TrackD0DiffEW = std::abs(Variable::trackD0(iPart) - Variable::trackD0(jPart)) / sqrt(pow(Variable::trackD0Error(
                      iPart), 2) + pow(Variable::trackD0Error(jPart), 2));
 
  m_TrackZ0DiffEW = std::abs(Variable::trackZ0(iPart) - Variable::trackZ0(jPart)) / sqrt(pow(Variable::trackZ0Error(
                      iPart), 2) + pow(Variable::trackZ0Error(jPart), 2));
 
  m_TrackTanLambdaDiffEW = std::abs(Variable::trackTanLambda(iPart) - Variable::trackTanLambda(jPart)) / sqrt(pow(
                             Variable::trackTanLambdaError(
                               iPart), 2) + pow(Variable::trackTanLambdaError(jPart), 2));
 
  m_TrackPhi0DiffEW = std::abs(Variable::trackPhi0(iPart) - Variable::trackPhi0(jPart)) / sqrt(pow(Variable::trackPhi0Error(
                        iPart), 2) + pow(Variable::trackPhi0Error(jPart), 2));
 
  m_TrackOmegaDiffEW = std::abs(Variable::trackOmega(iPart) - Variable::trackOmega(jPart)) / sqrt(pow(Variable::trackOmegaError(
                         iPart), 2) + pow(Variable::trackOmegaError(jPart), 2));
}

Member Data Documentation

m_ChargeProduct

Float_t m_ChargeProduct

private

charge(p1) * charge(p2)

Definition at line 112 of file SelectorMVA.h.

m_datafiles

std::vector<std::string> m_datafiles

private

name of datafile

Definition at line 98 of file SelectorMVA.h.

m_expert

MVA::FastBDTExpert m_expert

private

mva expert

Definition at line 91 of file SelectorMVA.h.

m_generalOptions

MVA::GeneralOptions m_generalOptions

private

mva general options (for the expert)

Definition at line 88 of file SelectorMVA.h.

m_identifier

std::string m_identifier

private

mva identifier

Definition at line 95 of file SelectorMVA.h.

m_IsCurl

Bool_t m_IsCurl

private

isCurl Truth

Definition at line 136 of file SelectorMVA.h.

m_PPhi

Float_t m_PPhi

private

angle between particle momentum vectors

Definition at line 109 of file SelectorMVA.h.

m_PtDiffEW

Float_t m_PtDiffEW

private

error weighted particle Pt difference

Definition at line 115 of file SelectorMVA.h.

m_PzDiffEW

Float_t m_PzDiffEW

private

error weighted particle Pz difference

Definition at line 118 of file SelectorMVA.h.

m_target_variable

std::string m_target_variable

private

name of target variable (isCurl)

Definition at line 104 of file SelectorMVA.h.

m_TFile

TFile* m_TFile

private

output file for training data

Definition at line 76 of file SelectorMVA.h.

m_TFileName

std::string m_TFileName

private

name of output file for training data

Definition at line 73 of file SelectorMVA.h.

m_TrackD0DiffEW

Float_t m_TrackD0DiffEW

private

error weighted track D0 difference

Definition at line 121 of file SelectorMVA.h.

m_TrackOmegaDiffEW

Float_t m_TrackOmegaDiffEW

private

error weighted track Omega difference

Definition at line 133 of file SelectorMVA.h.

m_TrackPhi0DiffEW

Float_t m_TrackPhi0DiffEW

private

error weighted track Phi0 difference

Definition at line 130 of file SelectorMVA.h.

m_TrackTanLambdaDiffEW

Float_t m_TrackTanLambdaDiffEW

private

error weighted track tan lambda diff difference

Definition at line 127 of file SelectorMVA.h.

m_TrackZ0DiffEW

Float_t m_TrackZ0DiffEW

private

error weighted track Z0 difference

Definition at line 124 of file SelectorMVA.h.

m_TrainFlag

bool m_TrainFlag

private

applying mva or training it

Definition at line 70 of file SelectorMVA.h.

m_TTree

TTree* m_TTree

private

training data tree

Definition at line 79 of file SelectorMVA.h.

m_variables

std::vector<std::string> m_variables

private

names of variables used by mva

Definition at line 101 of file SelectorMVA.h.

m_weightfile

MVA::Weightfile m_weightfile

private

mva weightfile

Definition at line 85 of file SelectorMVA.h.

m_weightfile_representation

std::unique_ptr<DBObjPtr<DatabaseRepresentationOfWeightfile> > m_weightfile_representation

private

Database pointer to the Database representation of the weightfile.

Definition at line 82 of file SelectorMVA.h.

---

The documentation for this class was generated from the following files:

• analysis/modules/CurlTagger/include/SelectorMVA.h
• analysis/modules/CurlTagger/src/SelectorMVA.cc