Skip to content

Commit 0447d2f

Browse files
authored
[ALICE3] Add more qa of ml use to otf multi-charm task (#12452)
1 parent 6e2cd14 commit 0447d2f

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

ALICE3/TableProducer/alice3-multicharmTable.cxx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ struct alice3multicharmTable {
124124
Configurable<float> xiccMaxEta{"xiccMaxEta", 1.5, "Max eta"};
125125
Configurable<float> massWindowXi{"massWindowXi", 0.015, "Mass window around Xi peak (GeV/c)"};
126126
Configurable<float> massWindowXiC{"massWindowXiC", 0.015, "Mass window around XiC peak (GeV/c)"};
127+
Configurable<float> massWindowXiCC{"massWindowXiCC", 0.4, "Mass window around XiCC peak (GeV/c). Make sure that bkg region is included in this window"};
127128

128129
ConfigurableAxis axisEta{"axisEta", {80, -4.0f, +4.0f}, "#eta"};
129130
ConfigurableAxis axisPt{"axisPt", {VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.4f, 4.8f, 5.2f, 5.6f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 17.0f, 19.0f, 21.0f, 23.0f, 25.0f, 30.0f, 35.0f, 40.0f, 50.0f}, "pt axis for QA histograms"};
@@ -251,6 +252,11 @@ struct alice3multicharmTable {
251252
}
252253

253254
thisXiCCcandidate.mass = RecoDecay::m(array{array{thisXiCCcandidate.prong0mom[0], thisXiCCcandidate.prong0mom[1], thisXiCCcandidate.prong0mom[2]}, array{thisXiCCcandidate.prong1mom[0], thisXiCCcandidate.prong1mom[1], thisXiCCcandidate.prong1mom[2]}}, array{mass0, mass1});
255+
256+
if (std::fabs(thisXiCCcandidate.mass - o2::constants::physics::MassXiCCPlusPlus) > massWindowXiCC) {
257+
return false;
258+
}
259+
254260
thisXiCCcandidate.pt = std::hypot(thisXiCCcandidate.prong0mom[0] + thisXiCCcandidate.prong1mom[0], thisXiCCcandidate.prong0mom[1] + thisXiCCcandidate.prong1mom[1]);
255261
thisXiCCcandidate.eta = RecoDecay::eta(array{thisXiCCcandidate.prong0mom[0] + thisXiCCcandidate.prong1mom[0], thisXiCCcandidate.prong0mom[1] + thisXiCCcandidate.prong1mom[1], thisXiCCcandidate.prong0mom[2] + thisXiCCcandidate.prong1mom[2]});
256262
return true;

ALICE3/Tasks/alice3-multicharm.cxx

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,8 @@ struct alice3multicharm {
252252

253253
histos.add("hBDTScore", "hBDTScore", kTH1D, {axisBDTScore});
254254
histos.add("hBDTScoreVsXiccMass", "hBDTScoreVsXiccMass", kTH2D, {axisXiccMass, axisBDTScore});
255-
histos.add("hBDTScoreVsXiccPt", "hBDTScoreVsXiccPt", kTH2D, {axisXiccMass, axisPt});
255+
histos.add("hBDTScoreVsXiccPt", "hBDTScoreVsXiccPt", kTH2D, {axisPt, axisBDTScore});
256+
histos.add("h3dBDTScore", "h3dBDTScore", kTH3D, {axisPt, axisXiccMass, axisBDTScore});
256257
for (const auto& score : bdt.requiredScores.value) {
257258
histPath = std::format("MLQA/RequiredBDTScore_{}/", static_cast<int>(score * 100));
258259
histPointers.insert({histPath + "hDCAXicDaughters", histos.add((histPath + "hDCAXicDaughters").c_str(), "hDCAXicDaughters", {kTH1D, {{axisDcaDaughters}}})});
@@ -293,7 +294,7 @@ struct alice3multicharm {
293294
}
294295

295296
template <typename TMCharmCands>
296-
void genericProcessXicc(TMCharmCands xiccCands)
297+
void genericProcessXicc(TMCharmCands const& xiccCands)
297298
{
298299
for (const auto& xiccCand : xiccCands) {
299300
if (bdt.enableML) {
@@ -324,6 +325,7 @@ struct alice3multicharm {
324325
histos.fill(HIST("hBDTScore"), bdtScore);
325326
histos.fill(HIST("hBDTScoreVsXiccMass"), xiccCand.xiccMass(), bdtScore);
326327
histos.fill(HIST("hBDTScoreVsXiccPt"), xiccCand.xiccPt(), bdtScore);
328+
histos.fill(HIST("h3dBDTScore"), xiccCand.xiccPt(), xiccCand.xiccMass(), bdtScore);
327329

328330
for (const auto& requiredScore : bdt.requiredScores.value) {
329331
if (bdtScore > requiredScore) {

0 commit comments

Comments
 (0)