From aa89e227337f2ab083d90dd56c1805d083b2a801 Mon Sep 17 00:00:00 2001 From: Marius Pfeiffer Date: Fri, 17 Nov 2023 13:18:07 +0100 Subject: [PATCH] some fitting adjustments --- inspect_tree.cpp | 24 +++++++++++++++++ train_bdt.cpp | 68 ++++++++++++++++++++++++++++++------------------ 2 files changed, 66 insertions(+), 26 deletions(-) diff --git a/inspect_tree.cpp b/inspect_tree.cpp index 487e40e..f94a440 100644 --- a/inspect_tree.cpp +++ b/inspect_tree.cpp @@ -24,6 +24,30 @@ int inspect_tree() { auto tree = file->Get("BuToKpMuMu23_noPID/DecayTree"); auto branches = tree->GetListOfBranches(); + // /*Lesen*/ + // Double_t Bu_M; + + // tree->SetBranchAddress("Bu_M", &Bu_M); + + // for (size_t i = 0; i < tree->GetEntries(); i++) + // { + // tree->GetEntry(i); + // // Bu_M verwenden + // } + + // /*Schreiben*/ + // Double_t B_Theta, B_Blubb; + + // tree->Branch("B_Theta", &B_Theta, "B_Theta/D"); + // tree->Branch("B_Blubb", &B_Blubb, "B_Blubb/D"); + + // for (size_t i = 0; i < 1000; i++) + // { + // B_Theta = 345435; // .... + // B_Blubb = 348535435435579; // .... + // tree->Fill(); + // } + for (size_t i = 0; i < branches->GetSize(); i++) { TBranch* b = (TBranch*) branches->At(i); diff --git a/train_bdt.cpp b/train_bdt.cpp index 255d9de..8733fa5 100644 --- a/train_bdt.cpp +++ b/train_bdt.cpp @@ -34,7 +34,7 @@ const bool TRAIN_TMVA = false; const bool EVALUATE_TMVA = true; -const int N_BINS = 150; +const int N_BINS = 130; const double B_PLUS_MASS = 5279.; const double J_PSI_MASS = 3096.; @@ -261,9 +261,9 @@ int train_bdt() // files to load std::vector mc_filenames = { - "/auto/data/pfeiffer/inclusive_detached_dilepton/MC/BuToKpMuMu_rd_btoxll_simulation_12143001_MagDown_v0r0p6316987.root"}; + "/auto/data/pfeiffer/inclusive_detached_dilepton/MC/BuToKpMuMu_rd_btoxll_simulation_12143001_MagDown_v0r0p6316365_FULLSTREAM.root"}; - TChain *mc_chain = new TChain("BuToKpMuMu23_noPID/DecayTree"); + TChain *mc_chain = new TChain("BuToKpMuMu_noPID/DecayTree"); for (unsigned int i = 0; i < mc_filenames.size(); i++) { mc_chain->Add(mc_filenames.at(i).c_str()); @@ -272,7 +272,9 @@ int train_bdt() std::vector vars{ TV::Float("Bplus_PT", "B_PT"), TV::Float("Bplus_BPVFDCHI2", "B_BPVFDCHI2"), + TV::Float("Bplus_BPVDIRA", "B_BPVDIRA"), TV::Float("Jpsi_BPVIPCHI2", "Jpsi_BPVIPCHI2"), + TV::Float("Jpsi_BPVDIRA", "Jpsi_BPVDIRA"), TV::Float("Jpsi_PT", "Jpsi_PT"), TV::Float("Kplus_BPVIPCHI2", "K_BPVIPCHI2"), TV::Float("Kplus_PT", "K_PT"), @@ -378,7 +380,7 @@ int train_bdt() sig_tree->Fill(); added_sig_entries++; - if (added_sig_entries >= added_bck_entries) { + if (added_sig_entries >= added_bck_entries * 2) { break; } } @@ -414,7 +416,7 @@ int train_bdt() data_loader->AddBackgroundTree(bkg_tree, background_weight); 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=500: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=600:MinNodeSize=2.5%:CreateMVAPdfs:MaxDepth=3:BoostType=AdaBoost:AdaBoostBeta=0.5:UseBaggedBoost:BaggedSampleFraction=0.5:SeparationType=GiniIndex:nCuts=20"); factory->TrainAllMethods(); factory->TestAllMethods(); @@ -477,7 +479,7 @@ int train_bdt() double mva_response = reader->EvaluateMVA("BDT"); h1_probs->Fill(mva_response); - const double mva_cut_value = 0.01; + const double mva_cut_value = 0.09; // -0.02; if (mva_response > mva_cut_value) { @@ -496,12 +498,12 @@ int train_bdt() c1->Draw(); - // TCanvas *c3 = new TCanvas("c3", "Canvas 3", 0, 0, 1000, 600); + TCanvas *c3 = new TCanvas("c3", "Canvas 3", 0, 0, 1000, 600); - // auto fitFrame = CreateRooFit(h1_Bplus_M); - // fitFrame->Draw(); + auto fitFrame = CreateRooFit(h1_Bplus_M); + fitFrame->Draw(); - // c3->Draw(); + c3->Draw(); } else { @@ -514,31 +516,45 @@ int train_bdt() RooPlot *CreateRooFit(TH1D *hist) { RooRealVar roo_var_mass("roo_var_mass", "B+ Mass Variable", 4700., 6500.); - roo_var_mass.setRange("fitting_range", 5000., 6000.); + roo_var_mass.setRange("fitting_range", 4800., 5800.); roo_var_mass.setRange("plot_range", 4700., 6500.); TString hist_name = "roohist_bplus_M"; RooDataHist roohist_bplus_M(hist_name, "B Plus Mass Histogram", roo_var_mass, RooFit::Import(*hist)); - // RooRealVar roo_sig_gauss_mean("roo_sig_gauss_mean", "Mass Gauss Mean", 5250., 5100., 5400.); - // RooRealVar roo_sig_gauss_sigma("roo_sig_gauss_sigma", "Mass Gauss Sigma", 60., 0., 150.); + RooRealVar roo_sig_gauss_mean("roo_sig_gauss_mean", "Mass Gauss Mean", 5250., 5100., 5400.); + RooRealVar roo_sig_gauss_sigma("roo_sig_gauss_sigma", "Mass Gauss Sigma", 30., 20., 40.); - // RooGaussian roo_sig_gauss("roo_sig_gauss", "B+ Mass Signal Gaussian", roo_var_mass, roo_sig_gauss_mean, roo_sig_gauss_sigma); + RooGaussian roo_sig_gauss("roo_sig_gauss", "B+ Mass Signal Gaussian", roo_var_mass, roo_sig_gauss_mean, roo_sig_gauss_sigma); // Crystal Ball for Signal - RooRealVar roo_sig_cb_x0("roo_sig_cry_x0", "Location", B_PLUS_MASS, 5100., 5400.); - RooRealVar roo_sig_cb_sigmaL("roo_sig_cry_sigmaL", "Sigma L", 30., 0., 60.); - RooRealVar roo_sig_cb_sigmaR("roo_sig_cry_sigmaR", "Sigma R", 30., 0., 60.); + // RooRealVar roo_sig_cb_x0("roo_sig_cry_x0", "Location", B_PLUS_MASS, 5100., 5400.); + // RooRealVar roo_sig_cb_sigmaL("roo_sig_cry_sigmaL", "Sigma L", 30., 0., 60.); + // RooRealVar roo_sig_cb_sigmaR("roo_sig_cry_sigmaR", "Sigma R", 30., 0., 60.); - RooRealVar roo_sig_cb_alphaL("roo_sig_cry_alphaL", "Alpha L", 15., 0., 30.); - RooRealVar roo_sig_cb_nL("roo_sig_cry_nL", "Exponent L", 10., 1., 20.); + // RooRealVar roo_sig_cb_alphaL("roo_sig_cry_alphaL", "Alpha L", 15., 0., 30.); + // RooRealVar roo_sig_cb_nL("roo_sig_cry_nL", "Exponent L", 0., -40., 40.); - RooRealVar roo_sig_cb_alphaR("roo_sig_cry_alphaR", "Alpha R", 15., 0., 30.); - RooRealVar roo_sig_cb_nR("roo_sig_cry_nR", "Exponent R", 10., 1., 20.); + // RooRealVar roo_sig_cb_alphaR("roo_sig_cry_alphaR", "Alpha R", 15., 0., 30.); + // RooRealVar roo_sig_cb_nR("roo_sig_cry_nR", "Exponent R", 0., -40., 40.); - RooCrystalBall roo_sig_cb("roo_sig_cb", "Signal Crystal Ball", roo_var_mass, roo_sig_cb_x0, roo_sig_cb_sigmaL, roo_sig_cb_sigmaR, roo_sig_cb_alphaL, roo_sig_cb_nL, roo_sig_cb_alphaR, roo_sig_cb_nR); + // RooCrystalBall roo_sig_cb("roo_sig_cb", "Signal Crystal Ball", roo_var_mass, roo_sig_cb_x0, roo_sig_cb_sigmaL, roo_sig_cb_sigmaR, roo_sig_cb_alphaL, roo_sig_cb_nL, roo_sig_cb_alphaR, roo_sig_cb_nR); - RooRealVar roo_bkg_exp_c("roo_bkg_exp_c", "Background C", -0.00099021, -0.003, -0.0001); + // Double Gauss Signal + // RooRealVar roo_sig_gauss1_mean("roo_sig_gauss1_mean", "Mass Gauss 1 Mean", 5250., 5200., 5300.); + // RooRealVar roo_sig_gauss1_sigma("roo_sig_gauss1_sigma", "Mass Gauss 1 Sigma", 60., 0., 150.); + + // RooGaussian roo_sig_gauss1("roo_sig_gauss1", "B+ Mass Signal 1 Gaussian", roo_var_mass, roo_sig_gauss1_mean, roo_sig_gauss1_sigma); + + // RooRealVar roo_sig_gauss2_mean("roo_sig_gauss2_mean", "Mass Gauss 2 Mean", 5250., 5200., 5300.); + // RooRealVar roo_sig_gauss2_sigma("roo_sig_gauss2_sigma", "Mass Gauss 2 Sigma", 60., 0., 150.); + + // RooGaussian roo_sig_gauss2("roo_sig_gauss2", "B+ Mass Signal 2 Gaussian", roo_var_mass, roo_sig_gauss2_mean, roo_sig_gauss2_sigma); + + // RooRealVar roo_sig_double_gauss_frac("roo_sig_double_gauss_frac", "B+ Mass Signal Double Gauss Frac", 0.5); + // RooAddPdf roo_sig_double_gauss("roo_sig_double_gauss", "B+ Mass Signal Double Gauss", roo_sig_gauss1, roo_sig_gauss2, roo_sig_double_gauss_frac); + + RooRealVar roo_bkg_exp_c("roo_bkg_exp_c", "Background C", -0.000693147, -0.002, 0.); RooExponential roo_bkg_exp("roo_bkg_exp", "B+ Mass Background Exp", roo_var_mass, roo_bkg_exp_c); RooRealVar roo_var_mass_nsig("roo_var_mass_nsig", "B+ Mass N Signal", 0., hist->GetEntries()); @@ -546,11 +562,11 @@ RooPlot *CreateRooFit(TH1D *hist) TString pdf_name = "roo_pdf_sig_plus_bkg"; RooAddPdf roo_pdf_sig_plus_bkg(pdf_name, "Sig + Bkg PDF", - RooArgList(roo_sig_cb, roo_bkg_exp), + RooArgList(roo_sig_gauss, roo_bkg_exp), RooArgList(roo_var_mass_nsig, roo_var_mass_nbkg)); RooPlot *roo_frame_mass = roo_var_mass.frame(); - roohist_bplus_M.plotOn(roo_frame_mass, RooFit::Binning(N_BINS), RooFit::Name(hist_name), RooFit::MarkerColor(18), RooFit::Range("plot_range")); + roohist_bplus_M.plotOn(roo_frame_mass, RooFit::Binning(N_BINS), RooFit::Name(hist_name), RooFit::MarkerColor(15), RooFit::Range("plot_range")); RooFitResult *fitres = roo_pdf_sig_plus_bkg.fitTo(roohist_bplus_M, RooFit::Save(), RooFit::PrintLevel(1), RooFit::Range("fitting_range")); @@ -561,7 +577,7 @@ RooPlot *CreateRooFit(TH1D *hist) // roo_sig_cb.plotOn(roo_frame_mass, RooFit::LineColor(kAlpine), RooFit::LineStyle(kDashed), RooFit::Range("fitting_range")); // roo_bkg_exp.plotOn(roo_frame_mass, RooFit::LineColor(kOrange), RooFit::LineStyle(kDashed), RooFit::Range("fitting_range")); - roo_sig_cb.paramOn(roo_frame_mass, RooFit::Layout(0.60, 0.99, 0.90)); + roo_sig_gauss.paramOn(roo_frame_mass, RooFit::Layout(0.60, 0.99, 0.90)); roo_frame_mass->getAttText()->SetTextSize(0.027); double ymax = roo_frame_mass->GetMaximum();