Skip to content

[PWGHF] Add an axis into THnSpares for ML study in the task #9814

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mfaggin merged 6 commits into from
Feb 7, 2025
Merged

[PWGHF] Add an axis into THnSpares for ML study in the task #9814

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
4e15d3d
Add an axis into the THnSparse to store the ML non-prompt score
JaeYoonCHO Feb 7, 2025
ad1036b
Fix O2linter warning
JaeYoonCHO Feb 7, 2025
57fa555
Fix O2linter warning
JaeYoonCHO Feb 7, 2025
cb24d0e
Please consider the following formatting changes
alibuild Feb 7, 2025
1e9b906
Merge pull request #5 from alibuild/alibot-cleanup-9814
JaeYoonCHO Feb 7, 2025
949ff8c
Merge branch 'AliceO2Group:master' into addNonPromptScore
JaeYoonCHO Feb 7, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 16 additions & 11 deletions PWGHF/D2H/Tasks/taskXicToXiPiPi.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,12 @@ struct HfTaskXicToXiPiPi {
Configurable<bool> checkDecayTypeMc{"checkDecayTypeMc", false, "Flag to enable DecayType histogram"};
// THnSparese for ML selection check
Configurable<bool> enableTHn{"enableTHn", false, "Fill THnSparse for Xic"};

Service<o2::framework::O2DatabasePDG> pdg;

Filter filterSelectCandidates = (aod::hf_sel_candidate_xic::isSelXicToXiPiPi >= selectionFlagXic);

// Axis
ConfigurableAxis thnConfigAxisPt{"thnConfigAxisPt", {400, 0., 40.}, ""};
ConfigurableAxis thnConfigAxisMass{"thnConfigAxisMass", {300, 1.8, 3.0}, ""};
ConfigurableAxis thnConfigAxisPtProng{"thnConfigAxisPtProng", {300, 0., 30.}, ""};
Expand All @@ -55,19 +61,15 @@ struct HfTaskXicToXiPiPi {
ConfigurableAxis thnConfigAxisDecLengthXY{"thnConfigAxisDecLengthXY", {200, 0., 0.5}, ""};
ConfigurableAxis thnConfigAxisCPA{"thnConfigAxisCPA", {110, -1.1, 1.1}, ""};
ConfigurableAxis thnConfigAxisBdtScoreBkg{"thnConfigAxisBdtScoreBkg", {100, 0., 1.}, ""};
ConfigurableAxis thnConfigAxisBdtScoreSignal{"thnConfigAxisBdtScoreSignal", {100, 0., 1.}, ""};
// Axis
ConfigurableAxis thnConfigAxisBdtScorePrompt{"thnConfigAxisBdtScorePrompt", {100, 0., 1.}, ""};
ConfigurableAxis thnConfigAxisBdtScoreNonPrompt{"thnConfigAxisBdtScoreNonPrompt", {100, 0., 1.}, ""};
ConfigurableAxis binsDecLength{"binsDecLength", {200, 0., 0.5}, ""};
ConfigurableAxis binsErrDecLength{"binsErrDecLength", {100, 0., 1.}, ""};
ConfigurableAxis binsDCA{"binsDCA", {100, -0.05, 0.05}, ""};
ConfigurableAxis binsImpParErr{"binsImpParErr", {200, -0.1, 0.1}, ""};
ConfigurableAxis binsSV{"binsSV", {200, -5., 5.}, ""};
ConfigurableAxis binsChi2{"binsChi2", {200, 0., 0.1}, ""};

Service<o2::framework::O2DatabasePDG> pdg;

Filter filterSelectCandidates = (aod::hf_sel_candidate_xic::isSelXicToXiPiPi >= selectionFlagXic);

HistogramRegistry registry{"registry"};

void init(InitContext const&)
Expand Down Expand Up @@ -251,11 +253,12 @@ struct HfTaskXicToXiPiPi {
const AxisSpec thnAxisDecLengthXY{thnConfigAxisDecLengthXY, "decay length xy (cm)"};
const AxisSpec thnAxisCPA{thnConfigAxisCPA, "#Xi^{#plus}_{c} candidate cosine of pointing angle"};
const AxisSpec thnAxisBdtScoreBkg{thnConfigAxisBdtScoreBkg, "BDT score of background"};
const AxisSpec thnAxisBdtScoreSignal{thnConfigAxisBdtScoreSignal, "BDT score of prompt Xic"};
const AxisSpec thnAxisBdtScorePrompt{thnConfigAxisBdtScorePrompt, "BDT score of prompt #Xi^{#plus}_{c}"};
const AxisSpec thnAxisBdtScoreNonPrompt{thnConfigAxisBdtScoreNonPrompt, "BDT score of non-prompt #Xi^{#plus}_{c}"};

if (doprocessWithKFParticleAndML || doprocessWithDCAFitterAndML || doprocessMcWithKFParticleAndML || doprocessMcWithDCAFitterAndML) {
// with ML information
registry.add("hXicToXiPiPiVarsWithML", "THnSparse for Xic with ML", HistType::kTHnSparseF, {thnAxisPt, thnAxisMass, thnAxisChi2PCA, thnAxisDecLength, thnAxisDecLengthXY, thnAxisCPA, thnAxisBdtScoreBkg, thnAxisBdtScoreSignal});
registry.add("hXicToXiPiPiVarsWithML", "THnSparse for Xic with ML", HistType::kTHnSparseF, {thnAxisPt, thnAxisMass, thnAxisChi2PCA, thnAxisDecLength, thnAxisDecLengthXY, thnAxisCPA, thnAxisBdtScoreBkg, thnAxisBdtScorePrompt, thnAxisBdtScoreNonPrompt});
} else {
// without ML information
registry.add("hXicToXiPiPiVars", "THnSparse for Xic", HistType::kTHnSparseF, {thnAxisPt, thnAxisMass, thnAxisChi2PCA, thnAxisDecLength, thnAxisDecLengthXY, thnAxisCPA});
Expand All @@ -274,13 +277,15 @@ struct HfTaskXicToXiPiPi {

if constexpr (useMl) {
// with ML information
double outputBkg = -99.;
double outputPrompt = -99.;
double outputBkg = -99.; // bkg score
double outputPrompt = -99.; // prompt score
double outputFD = -99.; // non-prompt score
if (candidate.mlProbXicToXiPiPi().size() > 0) {
outputBkg = candidate.mlProbXicToXiPiPi()[0];
outputPrompt = candidate.mlProbXicToXiPiPi()[1];
outputFD = candidate.mlProbXicToXiPiPi()[2];
}
registry.get<THnSparse>(HIST("hXicToXiPiPiVarsWithML"))->Fill(candidate.pt(), candidate.invMassXicPlus(), candidate.chi2PCA(), candidate.decayLength(), candidate.decayLengthXY(), candidate.cpa(), outputBkg, outputPrompt);
registry.get<THnSparse>(HIST("hXicToXiPiPiVarsWithML"))->Fill(candidate.pt(), candidate.invMassXicPlus(), candidate.chi2PCA(), candidate.decayLength(), candidate.decayLengthXY(), candidate.cpa(), outputBkg, outputPrompt, outputFD);
} else {
// without ML information
registry.get<THnSparse>(HIST("hXicToXiPiPiVars"))->Fill(candidate.pt(), candidate.invMassXicPlus(), candidate.chi2PCA(), candidate.decayLength(), candidate.decayLengthXY(), candidate.cpa());
Expand Down
Loading