 |
Belle II Software
release-05-01-25
|
13 #include <analysis/modules/CurlTagger/Selector.h>
15 #include <analysis/dataobjects/Particle.h>
19 #include <mva/methods/FastBDT.h>
30 namespace CurlTagger {
33 class SelectorMVA :
public Selector {
Float_t m_ChargeProduct
charge(p1) * charge(p2)
SelectorMVA(bool belleFlag, bool trainFlag)
Constructor.
std::string m_TFileName
name of output file for training data
virtual void initialize() override
initialize whatever needs to be initalized (root file etc)
Expert for the FastBDT MVA method.
Float_t m_TrackPhi0DiffEW
error weighted track Phi0 difference
virtual void finalize() override
finalize whatever needs to be finalized (train the MVA)
Float_t m_TrackD0DiffEW
error weighted track D0 difference
~SelectorMVA()
Destructor.
Float_t m_TrackZ0DiffEW
error weighted track Z0 difference
TFile * m_TFile
output file for training data
std::vector< std::string > m_datafiles
name of datafile
Abstract base class for different kinds of events.
void updateVariables(Particle *iPart, Particle *jPart)
updates the value of the MVA variable
MVA::FastBDTExpert m_expert
mva expert
Float_t m_PzDiffEW
error weighted particle Pz difference
Float_t m_TrackOmegaDiffEW
error weighted track Omega difference
virtual void collectTrainingInfo(Particle *iPart, Particle *jPart) override
collect training data and save to a root file
General options which are shared by all MVA trainings.
Float_t m_PPhi
angle between particle momentum vectors
bool m_TrainFlag
applying mva or training it
Class to store reconstructed particles.
MVA::GeneralOptions m_generalOptions
mva general options (for the expert)
std::string m_identifier
mva identifier
virtual float getResponse(Particle *iPart, Particle *jPart) override
Selector response that this pair of particles come from the same mc/actual particle.
Float_t m_PtDiffEW
error weighted particle Pt difference
std::vector< std::string > m_variables
names of variables used by mva
TTree * m_TTree
training data tree
Float_t m_TrackTanLambdaDiffEW
error weighted track tan lambda diff difference
virtual std::vector< float > getVariables(Particle *iPart, Particle *jPart) override
returns vector of variables used by this selector.
std::string m_target_variable
name of target variable (isCurl)
Float_t m_IsCurl
isCurl Truth