pfeiffer/inclusive_detached_dilepton
1
0
Fork 0
Code Issues Pull Requests 1 repo.project_board Releases Wiki Activity

try new training variable and pre filter setup

Browse Source
This commit is contained in:
Marius Pfeiffer 2024-04-10 11:14:56 +02:00
parent 06a90384eb
commit a4ed45d4ce
7 changed files with 159 additions and 121 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
B0ToHpHmMuMu_results.txt
View File

@ -1,11 +1,11 @@
#### B0ToHpHmMuMu @ Tue Apr 2 10:44:44 2024 #### #### B0ToHpHmMuMu @ Fri Apr 5 14:10:32 2024 ####
J/Psi Mode: 341 #pm 23 / 185 #pm 23 J/Psi Mode: 252 #pm 19 / 59 #pm 14
Sig/Bkg: 1.85 #pm 0.26 Sig/Bkg: 4.25 #pm 1.02
Psi(2S) Mode: 23 #pm 6 / 43 #pm 11 Psi(2S) Mode: 15 #pm 4 / 3 #pm 4
Sig/Bkg: 0.54 #pm 0.19 Sig/Bkg: 4.59 #pm 4.45
Mode Yield Ratio: 0.068 #pm 0.018 Mode Yield Ratio: 0.060 #pm 0.017
Rel Br Frac MuMu: 7.701 #pm 0.804 Rel Br Frac MuMu: 7.701 #pm 0.804
Rel Br Frac: 0.522 #pm 0.149 Rel Br Frac: 0.463 #pm 0.136
Fitted Parameters: Simulation Fitted Parameters: Simulation
@ -19,14 +19,14 @@ n_{R} = 5.67 \pm 0.73
Fitted Parameters: J/PSI Fitted Parameters: J/PSI
N_{Sig} = 341 \pm 23 N_{Sig} = 252 \pm 19
N_{Bkg} = 185 \pm 23 N_{Bkg} = 59 \pm 14
N_{Bkg,2#sigma} = 97 \pm 6 N_{Bkg,2#sigma} = 79 \pm 6
N_{Sig}/N_{Bkg} = 1.85 \pm 0.26 N_{Sig}/N_{Bkg} = 4.25 \pm 1.02
N_{Sig}/N_{Bkg,2#sigma} = 3.53 \pm 0.32 N_{Sig}/N_{Bkg,2#sigma} = 3.19 \pm 0.33
#lambda = -0.00217 \pm 0.00066 #lambda = -0.00400 \pm 0.00395
#mu = 5270.03 \pm 1.92 #mu = 5267.82 \pm 2.42
#sigma_{LR} = 26.26 \pm 1.98 #sigma_{LR} = 29.94 \pm 2.37
#alpha_{L} = 1.72 (c) #alpha_{L} = 1.72 (c)
n_{L} = 3.34 (c) n_{L} = 3.34 (c)
#alpha_{R} = 1.76 (c) #alpha_{R} = 1.76 (c)
@ -34,14 +34,14 @@ n_{R} = 5.67 (c)
Fitted Parameters: PSI(2S) Fitted Parameters: PSI(2S)
N_{Sig} = 23 \pm 6 N_{Sig} = 15 \pm 4
N_{Bkg} = 43 \pm 10 N_{Bkg} = 3 \pm 3
N_{Bkg,2#sigma} = 12 \pm 2 N_{Bkg,2#sigma} = 2 \pm 1
N_{Sig}/N_{Bkg} = 0.54 \pm 0.19 N_{Sig}/N_{Bkg} = 4.59 \pm 4.44
N_{Sig}/N_{Bkg,2#sigma} = 1.93 \pm 0.60 N_{Sig}/N_{Bkg,2#sigma} = 6.17 \pm 2.34
#lambda = -0.00202 \pm 0.00120 #lambda = -0.00000 \pm 0.00320
#mu = 5267.35 \pm 7.06 #mu = 5275.52 \pm 8.60
#sigma_{LR} = 26.26 (c) #sigma_{LR} = 29.94 (c)
#alpha_{L} = 1.72 (c) #alpha_{L} = 1.72 (c)
n_{L} = 3.34 (c) n_{L} = 3.34 (c)
#alpha_{R} = 1.76 (c) #alpha_{R} = 1.76 (c)
@ -50,11 +50,11 @@ n_{R} = 5.67 (c)
# J/psi # J/psi
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$341 \pm 23$ & $185 \pm 23$ $252 \pm 19$ & $59 \pm 14$
\end{tabular} \end{tabular}
# psi(2S) # psi(2S)
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$23 \pm 6$ & $43 \pm 10$ $15 \pm 4$ & $3 \pm 3$
\end{tabular} \end{tabular}

52
BuToHpMuMu_results.txt
View File

@ -1,18 +1,18 @@
#### BuToHpMuMu @ Tue Apr 2 10:44:14 2024 #### #### BuToHpMuMu @ Wed Apr 10 11:06:16 2024 ####
J/Psi Mode: 897 #pm 34 / 402 #pm 30 J/Psi Mode: 733 #pm 30 / 174 #pm 20
Sig/Bkg: 2.23 #pm 0.19 Sig/Bkg: 4.20 #pm 0.49
Psi(2S) Mode: 69 #pm 12 / 233 #pm 26 Psi(2S) Mode: 64 #pm 9 / 43 #pm 11
Sig/Bkg: 0.30 #pm 0.06 Sig/Bkg: 1.50 #pm 0.42
Mode Yield Ratio: 0.077 #pm 0.013 Mode Yield Ratio: 0.087 #pm 0.013
Rel Br Frac MuMu: 7.701 #pm 0.804 Rel Br Frac MuMu: 7.701 #pm 0.804
Rel Br Frac: 0.594 #pm 0.116 Rel Br Frac: 0.673 #pm 0.121
Params from Sim: Params from Sim:
aL: 1.76 aL: 1.76
nL: 4.02 nL: 4.02
aR: 1.93 aR: 1.93
nR: 4.33 nR: 4.33
S: 23.48 S: 23.72
#mu = 5278.16 \pm 0.05 #mu = 5278.16 \pm 0.05
#sigma_{LR} = 16.43 \pm 0.05 #sigma_{LR} = 16.43 \pm 0.05
@ -23,14 +23,14 @@ n_{R} = 4.33 \pm 0.19
Fitted Parameters: J/PSI Fitted Parameters: J/PSI
N_{Sig} = 897 \pm 34 N_{Sig} = 733 \pm 29
N_{Bkg} = 402 \pm 30 N_{Bkg} = 174 \pm 19
N_{Bkg,2#sigma} = 246 \pm 9 N_{Bkg,2#sigma} = 179 \pm 7
N_{Sig}/N_{Bkg} = 2.23 \pm 0.19 N_{Sig}/N_{Bkg} = 4.20 \pm 0.49
N_{Sig}/N_{Bkg,2#sigma} = 3.64 \pm 0.19 N_{Sig}/N_{Bkg,2#sigma} = 4.08 \pm 0.23
#lambda = -0.00350 \pm 0.00041 #lambda = -0.00400 \pm 0.00018
#mu = 5271.03 \pm 0.96 #mu = 5271.39 \pm 1.03
#sigma_{LR} = 23.48 \pm 0.90 #sigma_{LR} = 23.72 \pm 0.94
#alpha_{L} = 1.76 (c) #alpha_{L} = 1.76 (c)
n_{L} = 4.02 (c) n_{L} = 4.02 (c)
#alpha_{R} = 1.93 (c) #alpha_{R} = 1.93 (c)
@ -38,14 +38,14 @@ n_{R} = 4.33 (c)
Fitted Parameters: PSI(2S) Fitted Parameters: PSI(2S)
N_{Sig} = 69 \pm 11 N_{Sig} = 64 \pm 9
N_{Bkg} = 233 \pm 26 N_{Bkg} = 43 \pm 10
N_{Bkg,2#sigma} = 44 \pm 4 N_{Bkg,2#sigma} = 18 \pm 2
N_{Sig}/N_{Bkg} = 0.30 \pm 0.06 N_{Sig}/N_{Bkg} = 1.50 \pm 0.42
N_{Sig}/N_{Bkg,2#sigma} = 1.56 \pm 0.29 N_{Sig}/N_{Bkg,2#sigma} = 3.56 \pm 0.67
#lambda = -0.00142 \pm 0.00056 #lambda = -0.00217 \pm 0.00121
#mu = 5263.32 \pm 4.14 #mu = 5264.03 \pm 3.67
#sigma_{LR} = 23.48 (c) #sigma_{LR} = 23.72 (c)
#alpha_{L} = 1.76 (c) #alpha_{L} = 1.76 (c)
n_{L} = 4.02 (c) n_{L} = 4.02 (c)
#alpha_{R} = 1.93 (c) #alpha_{R} = 1.93 (c)
@ -54,11 +54,11 @@ n_{R} = 4.33 (c)
# J/psi # J/psi
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$897 \pm 34$ & $402 \pm 30$ $733 \pm 29$ & $174 \pm 19$
\end{tabular} \end{tabular}
# psi(2S) # psi(2S)
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$69 \pm 11$ & $233 \pm 26$ $64 \pm 9$ & $43 \pm 10$
\end{tabular} \end{tabular}

44
BuToKpMuMu_results.txt
View File

@ -1,7 +1,7 @@
#### BuToKpMuMu @ Tue Apr 2 10:45:35 2024 #### #### BuToKpMuMu @ Wed Apr 10 10:51:08 2024 ####
J/Psi Mode: 815 #pm 35 / 743 #pm 45 J/Psi Mode: 813 #pm 35 / 734 #pm 44
Sig/Bkg: 1.10 #pm 0.09 Sig/Bkg: 1.11 #pm 0.09
Psi(2S) Mode: 69 #pm 12 / 290 #pm 30 Psi(2S) Mode: 69 #pm 12 / 283 #pm 30
Sig/Bkg: 0.24 #pm 0.05 Sig/Bkg: 0.24 #pm 0.05
Mode Yield Ratio: 0.085 #pm 0.015 Mode Yield Ratio: 0.085 #pm 0.015
Rel Br Frac MuMu: 7.701 #pm 0.804 Rel Br Frac MuMu: 7.701 #pm 0.804
@ -12,7 +12,7 @@ aL: 1.77
nL: 3.86 nL: 3.86
aR: 1.96 aR: 1.96
nR: 4.43 nR: 4.43
S: 24.12 S: 24.08
#mu = 5278.10 \pm 0.05 #mu = 5278.10 \pm 0.05
#sigma_{LR} = 16.35 \pm 0.05 #sigma_{LR} = 16.35 \pm 0.05
@ -23,14 +23,14 @@ n_{R} = 4.43 \pm 0.20
Fitted Parameters: J/PSI Fitted Parameters: J/PSI
N_{Sig} = 815 \pm 34 N_{Sig} = 813 \pm 34
N_{Bkg} = 743 \pm 44 N_{Bkg} = 734 \pm 44
N_{Bkg,2#sigma} = 259 \pm 9 N_{Bkg,2#sigma} = 257 \pm 9
N_{Sig}/N_{Bkg} = 1.10 \pm 0.08 N_{Sig}/N_{Bkg} = 1.11 \pm 0.08
N_{Sig}/N_{Bkg,2#sigma} = 3.15 \pm 0.18 N_{Sig}/N_{Bkg,2#sigma} = 3.16 \pm 0.18
#lambda = -0.00207 \pm 0.00031 #lambda = -0.00209 \pm 0.00031
#mu = 5271.47 \pm 1.08 #mu = 5271.35 \pm 1.08
#sigma_{LR} = 24.12 \pm 1.02 #sigma_{LR} = 24.08 \pm 1.02
#alpha_{L} = 1.77 (c) #alpha_{L} = 1.77 (c)
n_{L} = 3.86 (c) n_{L} = 3.86 (c)
#alpha_{R} = 1.96 (c) #alpha_{R} = 1.96 (c)
@ -38,14 +38,14 @@ n_{R} = 4.43 (c)
Fitted Parameters: PSI(2S) Fitted Parameters: PSI(2S)
N_{Sig} = 69 \pm 12 N_{Sig} = 69 \pm 11
N_{Bkg} = 290 \pm 30 N_{Bkg} = 283 \pm 29
N_{Bkg,2#sigma} = 53 \pm 5 N_{Bkg,2#sigma} = 52 \pm 5
N_{Sig}/N_{Bkg} = 0.24 \pm 0.05 N_{Sig}/N_{Bkg} = 0.24 \pm 0.05
N_{Sig}/N_{Bkg,2#sigma} = 1.32 \pm 0.25 N_{Sig}/N_{Bkg,2#sigma} = 1.33 \pm 0.25
#lambda = -0.00130 \pm 0.00051 #lambda = -0.00136 \pm 0.00051
#mu = 5268.10 \pm 4.25 #mu = 5268.31 \pm 4.17
#sigma_{LR} = 24.12 (c) #sigma_{LR} = 24.08 (c)
#alpha_{L} = 1.77 (c) #alpha_{L} = 1.77 (c)
n_{L} = 3.86 (c) n_{L} = 3.86 (c)
#alpha_{R} = 1.96 (c) #alpha_{R} = 1.96 (c)
@ -54,11 +54,11 @@ n_{R} = 4.43 (c)
# J/psi # J/psi
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$815 \pm 34$ & $743 \pm 44$ $813 \pm 34$ & $734 \pm 44$
\end{tabular} \end{tabular}
# psi(2S) # psi(2S)
\begin{tabular}{c|c} \begin{tabular}{c|c}
$N_{Sig}$ & $N_{Bkg}$\\\hline $N_{Sig}$ & $N_{Bkg}$\\\hline
$69 \pm 12$ & $290 \pm 30$ $69 \pm 11$ & $283 \pm 29$
\end{tabular} \end{tabular}

2
bdt_classification.h
View File

@ -229,7 +229,7 @@ void TrainBDT(std::vector<TV *> vars, const char* unique_id, TTree *sig_tree, TT
data_loader->AddBackgroundTree(bkg_tree, background_weight); data_loader->AddBackgroundTree(bkg_tree, background_weight);
data_loader->PrepareTrainingAndTestTree("", "", "nTrain_Signal=0:nTrain_Background=0:SplitMode=Random:NormMode=NumEvents:V"); data_loader->PrepareTrainingAndTestTree("", "", "nTrain_Signal=0:nTrain_Background=0:SplitMode=Random:NormMode=NumEvents:V");
factory->BookMethod(data_loader, TMVA::Types::kBDT, "BDT", "!H:!V:NTrees=600:MinNodeSize=2.5%:CreateMVAPdfs:MaxDepth=3:BoostType=AdaBoost:AdaBoostBeta=0.5:UseBaggedBoost:BaggedSampleFraction=0.5:SeparationType=GiniIndex:nCuts=20"); factory->BookMethod(data_loader, TMVA::Types::kBDT, "BDT", "!H:!V:NTrees=400:MinNodeSize=2.5%:CreateMVAPdfs:MaxDepth=3:BoostType=AdaBoost:AdaBoostBeta=0.5:UseBaggedBoost:BaggedSampleFraction=0.5:SeparationType=GiniIndex:nCuts=20");
factory->TrainAllMethods(); factory->TrainAllMethods();
factory->TestAllMethods(); factory->TestAllMethods();

19
new_analysis_b02hphmmumu.cpp
View File

@ -51,6 +51,7 @@ int new_analysis_b02hphmmumu()
const char *sim_tree_name = "B0ToHpHmMuMu_noPID_mapped"; const char *sim_tree_name = "B0ToHpHmMuMu_noPID_mapped";
const char *end_state_mass_literal = "m(#pi^{+}#pi^{-}_{(#rightarrow K^{-})}#mu^{+}#mu^{-} & #pi^{+}_{(#rightarrow K^{+})}#pi^{-}#mu^{+}#mu^{-})"; const char *end_state_mass_literal = "m(#pi^{+}#pi^{-}_{(#rightarrow K^{-})}#mu^{+}#mu^{-} & #pi^{+}_{(#rightarrow K^{+})}#pi^{-}#mu^{+}#mu^{-})";
const bool retrain_bdt = false; const bool retrain_bdt = false;
const bool skip_fit = false;
TChain *data_chain = new TChain(TString::Format("%s/DecayTree", data_tree_name)); TChain *data_chain = new TChain(TString::Format("%s/DecayTree", data_tree_name));
data_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/data_samples/Collision23_Beam6800GeV-VeloClosed-MagDown-Excl-UT_RealData_Sprucing23r1_90000000_RD.root"); data_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/data_samples/Collision23_Beam6800GeV-VeloClosed-MagDown-Excl-UT_RealData_Sprucing23r1_90000000_RD.root");
@ -108,8 +109,10 @@ int new_analysis_b02hphmmumu()
TV::Float("B0_PT", "B0_PT"), TV::Float("B0_PT", "B0_PT"),
TV::Float("B0_BPVFDCHI2", "B0_BPVFDCHI2"), TV::Float("B0_BPVFDCHI2", "B0_BPVFDCHI2"),
TV::Float("B0_BPVDIRA", "B0_BPVDIRA"), TV::Float("B0_BPVDIRA", "B0_BPVDIRA"),
TV::Double("B0_CHI2", "B0_CHI2"),
TV::Float("Jpsi_BPVIPCHI2", "Jpsi_BPVIPCHI2"), TV::Float("Jpsi_BPVIPCHI2", "Jpsi_BPVIPCHI2"),
TV::Float("Jpsi_PT", "Jpsi_PT"), TV::Float("Jpsi_PT", "Jpsi_PT"),
TV::Double("Jpsi_CHI2", "Jpsi_CHI2"),
TV::Float("Kst0_BPVIPCHI2", "Kst0_BPVIPCHI2"), TV::Float("Kst0_BPVIPCHI2", "Kst0_BPVIPCHI2"),
TV::Float("Kst0_PT", "Kst0_PT"), TV::Float("Kst0_PT", "Kst0_PT"),
TV::Float("Kplus_BPVIPCHI2", "Kplus_BPVIPCHI2"), TV::Float("Kplus_BPVIPCHI2", "Kplus_BPVIPCHI2"),
@ -117,10 +120,10 @@ int new_analysis_b02hphmmumu()
TV::Float("piminus_BPVIPCHI2", "piminus_BPVIPCHI2"), TV::Float("piminus_BPVIPCHI2", "piminus_BPVIPCHI2"),
TV::Float("piminus_PT", "piminus_PT"), TV::Float("piminus_PT", "piminus_PT"),
kplus_pid_k_var, kplus_pid_k_var,
TV::Float("muminus_BPVIPCHI2", "muminus_BPVIPCHI2"), //TV::Float("muminus_BPVIPCHI2", "muminus_BPVIPCHI2"),
// TV::Float("muminus_PT", "muminus_PT"), TV::Float("muminus_PT", "muminus_PT"),
TV::Float("muplus_BPVIPCHI2", "muplus_BPVIPCHI2"), //TV::Float("muplus_BPVIPCHI2", "muplus_BPVIPCHI2"),
// TV::Float("muplus_PT", "muplus_PT"), TV::Float("muplus_PT", "muplus_PT"),
}; };
TTree *sig_tree = new TTree("TreeS", "tree containing signal data"); TTree *sig_tree = new TTree("TreeS", "tree containing signal data");
@ -147,7 +150,8 @@ int new_analysis_b02hphmmumu()
if (std::all_of(vars.begin(), vars.end(), [](TV *v) if (std::all_of(vars.begin(), vars.end(), [](TV *v)
{ return v->IsDataFinite(); })) { return v->IsDataFinite(); }))
{ {
if (reconstructed_B_Mass > 5500. && ((TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) || (TMath::Abs(dimuon.M() - PSI2S_MASS) < 100.))) if (reconstructed_B_Mass > 5500. && ((TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) || (TMath::Abs(dimuon.M() - PSI2S_MASS) < 100.))
&& kplus_pid_k_var->GetDataDouble() > -3 && (kplus_pid_k_var->GetDataDouble() - piminus_PID_K) > 0)
{ {
bkg_tree->Fill(); bkg_tree->Fill();
bkg_events++; bkg_events++;
@ -194,6 +198,11 @@ int new_analysis_b02hphmmumu()
std::cout << "# Finished BDT retrain." << std::endl; std::cout << "# Finished BDT retrain." << std::endl;
} }
if (skip_fit) {
std::cout << "# Skipping evaluation of data." << std::endl;
return 0;
}
std::cout << "# Starting evaluation of data." << std::endl; std::cout << "# Starting evaluation of data." << std::endl;
Float_t *train_vars = new Float_t[vars.size()]; Float_t *train_vars = new Float_t[vars.size()];

41
new_analysis_bu2hpmumu.cpp
View File

@ -50,7 +50,8 @@ int new_analysis_bu2hpmumu()
const char *data_tree_name = "SpruceRD_BuToHpMuMu"; const char *data_tree_name = "SpruceRD_BuToHpMuMu";
const char *sim_tree_name = "BuToHpMuMu_noPID_mapped"; const char *sim_tree_name = "BuToHpMuMu_noPID_mapped";
const char *end_state_mass_literal = "m(#pi^{+}_{(#rightarrow K^{+})}#mu^{+}#mu^{-})"; const char *end_state_mass_literal = "m(#pi^{+}_{(#rightarrow K^{+})}#mu^{+}#mu^{-})";
const bool retrain_bdt = false; const bool retrain_bdt = true;
const bool skip_fit = false;
TChain *data_chain = new TChain(TString::Format("%s/DecayTree", data_tree_name)); TChain *data_chain = new TChain(TString::Format("%s/DecayTree", data_tree_name));
data_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/data_samples/Collision23_Beam6800GeV-VeloClosed-MagDown-Excl-UT_RealData_Sprucing23r1_90000000_RD.root"); data_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/data_samples/Collision23_Beam6800GeV-VeloClosed-MagDown-Excl-UT_RealData_Sprucing23r1_90000000_RD.root");
@ -60,9 +61,9 @@ int new_analysis_bu2hpmumu()
FourVect *l24v_data = FourVect::Init(data_chain, "muplus"); FourVect *l24v_data = FourVect::Init(data_chain, "muplus");
FourVect *hp4v_data = FourVect::Init(data_chain, "Kplus"); FourVect *hp4v_data = FourVect::Init(data_chain, "Kplus");
// Double_t Kplus_PID_K; Double_t Kplus_PID_K;
// data_chain->SetBranchAddress("Kplus_PID_K", &Kplus_PID_K); data_chain->SetBranchAddress("Kplus_PID_K", &Kplus_PID_K);
TChain *sim_chain = new TChain(TString::Format("%s/DecayTree", sim_tree_name)); TChain *sim_chain = new TChain(TString::Format("%s/DecayTree", sim_tree_name));
sim_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/MC/BuToHpMuMu_mapped_mc.root"); sim_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/MC/BuToHpMuMu_mapped_mc.root");
@ -100,17 +101,19 @@ int new_analysis_bu2hpmumu()
auto hlt1_decision_histos = CreateHlt1DecisionHistos(analysis_name); auto hlt1_decision_histos = CreateHlt1DecisionHistos(analysis_name);
std::map<std::string, int> exclusive_hits{}; std::map<std::string, int> exclusive_hits{};
TV* kplus_pid_k_var = TV::Double("Kplus_PID_K", "Kplus_PID_K"); // TV* kplus_pid_k_var = TV::Double("Kplus_PID_K", "Kplus_PID_K");
std::vector<TV *> vars{ std::vector<TV *> vars{
TV::Float("B_PT", "B_PT"), TV::Float("B_PT", "B_PT"),
TV::Float("B_BPVFDCHI2", "B_BPVFDCHI2"), TV::Float("B_BPVFDCHI2", "B_BPVFDCHI2"),
TV::Float("B_BPVDIRA", "B_BPVDIRA"), TV::Float("B_BPVDIRA", "B_BPVDIRA"),
TV::Double("B_CHI2", "B_CHI2"),
TV::Float("Jpsi_BPVIPCHI2", "Jpsi_BPVIPCHI2"), TV::Float("Jpsi_BPVIPCHI2", "Jpsi_BPVIPCHI2"),
TV::Float("Jpsi_PT", "Jpsi_PT"), TV::Float("Jpsi_PT", "Jpsi_PT"),
TV::Double("Jpsi_CHI2", "Jpsi_CHI2"),
TV::Float("Kplus_BPVIPCHI2", "Kplus_BPVIPCHI2"), TV::Float("Kplus_BPVIPCHI2", "Kplus_BPVIPCHI2"),
TV::Float("Kplus_PT", "Kplus_PT"), TV::Float("Kplus_PT", "Kplus_PT"),
kplus_pid_k_var, // kplus_pid_k_var,
TV::Float("muminus_BPVIPCHI2", "muminus_BPVIPCHI2"), TV::Float("muminus_BPVIPCHI2", "muminus_BPVIPCHI2"),
// TV::Float("muminus_PT", "muminus_PT"), // TV::Float("muminus_PT", "muminus_PT"),
TV::Float("muplus_BPVIPCHI2", "muplus_BPVIPCHI2"), TV::Float("muplus_BPVIPCHI2", "muplus_BPVIPCHI2"),
@ -142,7 +145,7 @@ int new_analysis_bu2hpmumu()
if (std::all_of(vars.begin(), vars.end(), [](TV *v) if (std::all_of(vars.begin(), vars.end(), [](TV *v)
{ return v->IsDataFinite(); })) { return v->IsDataFinite(); }))
{ {
if (reconstructed_B_Mass > 5500. && ((TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) || (TMath::Abs(dimuon.M() - PSI2S_MASS) < 100.))) if (reconstructed_B_Mass > 5500. && ((TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) || (TMath::Abs(dimuon.M() - PSI2S_MASS) < 100.)) && Kplus_PID_K > -3)
{ {
bkg_tree->Fill(); bkg_tree->Fill();
bkg_events++; bkg_events++;
@ -186,6 +189,12 @@ int new_analysis_bu2hpmumu()
std::cout << "# Finished BDT retrain." << std::endl; std::cout << "# Finished BDT retrain." << std::endl;
} }
if (skip_fit)
{
std::cout << "# Skipping evaluation of data." << std::endl;
return 0;
}
std::cout << "# Starting evaluation of data." << std::endl; std::cout << "# Starting evaluation of data." << std::endl;
Float_t *train_vars = new Float_t[vars.size()]; Float_t *train_vars = new Float_t[vars.size()];
@ -222,12 +231,14 @@ int new_analysis_bu2hpmumu()
FillHlt1DecisionHistos(hlt1_decision_histos, reconstructed_B_Mass); FillHlt1DecisionHistos(hlt1_decision_histos, reconstructed_B_Mass);
} }
if (Kplus_PID_K > -3 && ((TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) || (TMath::Abs(dimuon.M() - PSI2S_MASS) < 100.)))
{
double mva_response = reader->EvaluateMVA("BDT"); double mva_response = reader->EvaluateMVA("BDT");
h1_bdt_probs->Fill(mva_response); h1_bdt_probs->Fill(mva_response);
h1_B_Mass_unf->Fill(reconstructed_B_Mass); h1_B_Mass_unf->Fill(reconstructed_B_Mass);
if (mva_response > mva_cut_value && kplus_pid_k_var->GetDataDouble() > -3) if (mva_response > mva_cut_value)
{ {
h1_B_Mass_bdtf->Fill(reconstructed_B_Mass); h1_B_Mass_bdtf->Fill(reconstructed_B_Mass);
if (TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) if (TMath::Abs(dimuon.M() - JPSI_MASS) < 100.)
@ -241,6 +252,7 @@ int new_analysis_bu2hpmumu()
tree_B_Mass_psi2s->Fill(); tree_B_Mass_psi2s->Fill();
} }
} }
}
PrintProgress(TString::Format("%s BDT Evaluation", analysis_name), data_entries, 10000, i); PrintProgress(TString::Format("%s BDT Evaluation", analysis_name), data_entries, 10000, i);
} }
@ -296,30 +308,35 @@ int new_analysis_bu2hpmumu()
res_file << "Rel Br Frac MuMu: " << ErrToStr(mumu_br_frac, 3) << std::endl; res_file << "Rel Br Frac MuMu: " << ErrToStr(mumu_br_frac, 3) << std::endl;
auto br_frac = MultWithErr(signal_ratio.first, signal_ratio.second, mumu_br_frac.first, mumu_br_frac.second); auto br_frac = MultWithErr(signal_ratio.first, signal_ratio.second, mumu_br_frac.first, mumu_br_frac.second);
res_file << "Rel Br Frac: " << ErrToStr(br_frac, 3) << std::endl << std::endl; res_file << "Rel Br Frac: " << ErrToStr(br_frac, 3) << std::endl
<< std::endl;
res_file << "Params from Sim:" << std::endl << roofit_hist_sim.shape_parameters.ToString() << std::endl; res_file << "Params from Sim:" << std::endl
<< roofit_hist_sim.shape_parameters.ToString() << std::endl;
for (const auto &par : roofit_hist_sim.fitted_params) for (const auto &par : roofit_hist_sim.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
res_file << std::endl << "Fitted Parameters: J/PSI" << std::endl << std::endl; res_file << std::endl
<< "Fitted Parameters: J/PSI" << std::endl
<< std::endl;
for (const auto &par : roofit_hist_jpsi_fitsum.fitted_params) for (const auto &par : roofit_hist_jpsi_fitsum.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
res_file << std::endl << "Fitted Parameters: PSI(2S)" << std::endl << std::endl; res_file << std::endl
<< "Fitted Parameters: PSI(2S)" << std::endl
<< std::endl;
for (const auto &par : roofit_hist_psi2s_fitsum.fitted_params) for (const auto &par : roofit_hist_psi2s_fitsum.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
auto print_table = [&res_file](std::string name, std::pair<double, double> sig, std::pair<double, double> bkg) auto print_table = [&res_file](std::string name, std::pair<double, double> sig, std::pair<double, double> bkg)
{ {
res_file << std::endl; res_file << std::endl;

22
new_analysis_bu2kpmumu.cpp
View File

@ -58,6 +58,10 @@ int new_analysis_bu2kpmumu()
FourVect *l24v_data = FourVect::Init(data_chain, "muminus"); FourVect *l24v_data = FourVect::Init(data_chain, "muminus");
FourVect *hp4v_data = FourVect::Init(data_chain, "Kplus"); FourVect *hp4v_data = FourVect::Init(data_chain, "Kplus");
Double_t Kplus_PID_K;
data_chain->SetBranchAddress("Kplus_PID_K", &Kplus_PID_K);
TChain *sim_chain = new TChain(TString::Format("%s/DecayTree", sim_tree_name)); TChain *sim_chain = new TChain(TString::Format("%s/DecayTree", sim_tree_name));
sim_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/MC/rd_btoxll_simulation_turbo_v0r0p6657752_BuToKpMuMu_12143001_magdown.root"); sim_chain->Add("/auto/data/pfeiffer/inclusive_detached_dilepton/MC/rd_btoxll_simulation_turbo_v0r0p6657752_BuToKpMuMu_12143001_magdown.root");
@ -134,6 +138,8 @@ int new_analysis_bu2kpmumu()
h1_B_Mass_unf->Fill(reconstructed_B_Mass); h1_B_Mass_unf->Fill(reconstructed_B_Mass);
if (Kplus_PID_K > -3)
{
if (TMath::Abs(dimuon.M() - JPSI_MASS) < 100.) if (TMath::Abs(dimuon.M() - JPSI_MASS) < 100.)
{ {
B_Mass_jpsi_var = reconstructed_B_Mass; B_Mass_jpsi_var = reconstructed_B_Mass;
@ -144,6 +150,7 @@ int new_analysis_bu2kpmumu()
B_Mass_psi2s_var = reconstructed_B_Mass; B_Mass_psi2s_var = reconstructed_B_Mass;
tree_B_Mass_psi2s->Fill(); tree_B_Mass_psi2s->Fill();
} }
}
PrintProgress(TString::Format("%s BDT Evaluation", analysis_name), data_entries, 10000, i); PrintProgress(TString::Format("%s BDT Evaluation", analysis_name), data_entries, 10000, i);
} }
@ -191,30 +198,35 @@ int new_analysis_bu2kpmumu()
res_file << "Rel Br Frac MuMu: " << ErrToStr(mumu_br_frac, 3) << std::endl; res_file << "Rel Br Frac MuMu: " << ErrToStr(mumu_br_frac, 3) << std::endl;
auto br_frac = MultWithErr(signal_ratio.first, signal_ratio.second, mumu_br_frac.first, mumu_br_frac.second); auto br_frac = MultWithErr(signal_ratio.first, signal_ratio.second, mumu_br_frac.first, mumu_br_frac.second);
res_file << "Rel Br Frac: " << ErrToStr(br_frac, 3) << std::endl << std::endl; res_file << "Rel Br Frac: " << ErrToStr(br_frac, 3) << std::endl
<< std::endl;
res_file << "Params from Sim:" << std::endl << roofit_hist_sim.shape_parameters.ToString() << std::endl; res_file << "Params from Sim:" << std::endl
<< roofit_hist_sim.shape_parameters.ToString() << std::endl;
for (const auto &par : roofit_hist_sim.fitted_params) for (const auto &par : roofit_hist_sim.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
res_file << std::endl << "Fitted Parameters: J/PSI" << std::endl << std::endl; res_file << std::endl
<< "Fitted Parameters: J/PSI" << std::endl
<< std::endl;
for (const auto &par : roofit_hist_jpsi_fitsum.fitted_params) for (const auto &par : roofit_hist_jpsi_fitsum.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
res_file << std::endl << "Fitted Parameters: PSI(2S)" << std::endl << std::endl; res_file << std::endl
<< "Fitted Parameters: PSI(2S)" << std::endl
<< std::endl;
for (const auto &par : roofit_hist_psi2s_fitsum.fitted_params) for (const auto &par : roofit_hist_psi2s_fitsum.fitted_params)
{ {
res_file << par.ToString(true).c_str() << std::endl; res_file << par.ToString(true).c_str() << std::endl;
} }
auto print_table = [&res_file](std::string name, std::pair<double, double> sig, std::pair<double, double> bkg) auto print_table = [&res_file](std::string name, std::pair<double, double> sig, std::pair<double, double> bkg)
{ {
res_file << std::endl; res_file << std::endl;