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

#include <SelectorMVA.h>

Inheritance diagram for SelectorMVA:

Collaboration diagram for SelectorMVA:

[legend]

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 18 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 71 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 133 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 120 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 143 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 63 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 77 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 111 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 31 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 = abs(Variable::particlePt(iPart) - Variable::particlePt(jPart)) / sqrt(pow(Variable::particlePtErr(
                 iPart), 2) + pow(Variable::particlePtErr(jPart), 2));
 
  m_PzDiffEW = abs(Variable::particlePz(iPart) - Variable::particlePz(jPart)) / sqrt(pow(Variable::particlePzErr(
                 iPart), 2) + pow(Variable::particlePzErr(jPart), 2));
 
  m_TrackD0DiffEW = abs(Variable::trackD0(iPart) - Variable::trackD0(jPart)) / sqrt(pow(Variable::trackD0Error(
                      iPart), 2) + pow(Variable::trackD0Error(jPart), 2));
 
  m_TrackZ0DiffEW = abs(Variable::trackZ0(iPart) - Variable::trackZ0(jPart)) / sqrt(pow(Variable::trackZ0Error(
                      iPart), 2) + pow(Variable::trackZ0Error(jPart), 2));
 
  m_TrackTanLambdaDiffEW = abs(Variable::trackTanLambda(iPart) - Variable::trackTanLambda(jPart)) / sqrt(pow(
                             Variable::trackTanLambdaError(
                               iPart), 2) + pow(Variable::trackTanLambdaError(jPart), 2));
 
  m_TrackPhi0DiffEW = abs(Variable::trackPhi0(iPart) - Variable::trackPhi0(jPart)) / sqrt(pow(Variable::trackPhi0Error(
                        iPart), 2) + pow(Variable::trackPhi0Error(jPart), 2));
 
  m_TrackOmegaDiffEW = 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

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ SelectorMVA()

Member Function Documentation

◆ collectTrainingInfo()

◆ finalize()

◆ getOptimalResponseCut()

◆ getResponse()

◆ getVariables()

◆ initialize()

◆ initializeMVA()

◆ updateVariables()

Member Data Documentation

◆ m_ChargeProduct

◆ m_datafiles

◆ m_expert

◆ m_generalOptions

◆ m_identifier

◆ m_IsCurl

◆ m_PPhi

◆ m_PtDiffEW

◆ m_PzDiffEW

◆ m_target_variable

◆ m_TFile

◆ m_TFileName

◆ m_TrackD0DiffEW

◆ m_TrackOmegaDiffEW

◆ m_TrackPhi0DiffEW

◆ m_TrackTanLambdaDiffEW

◆ m_TrackZ0DiffEW

◆ m_TrainFlag

◆ m_TTree

◆ m_variables

◆ m_weightfile

◆ m_weightfile_representation