Skip to content

Commit 96f9444

Browse files
[PWGHF] Fix D0 ML prompt and non-prompt score ordering (#17173)
1 parent d5e065d commit 96f9444

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

PWGHF/Core/HfMlResponseD0ToKPi.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,8 +225,8 @@ class HfMlResponseD0ToKPi : public HfMlResponse<TypeOutputScore>
225225
CHECK_AND_FILL_VEC_D0_SIGNED(candidate, nSigTpcTofKaExpKa, tpcTofNSigmaKa1, tpcTofNSigmaKa0);
226226

227227
CHECK_AND_FILL_VEC_D0_ML(candidate, bdtOutputBkg, mlProbD0, mlProbD0bar, 0);
228-
CHECK_AND_FILL_VEC_D0_ML(candidate, bdtOutputNonPrompt, mlProbD0, mlProbD0bar, 1);
229-
CHECK_AND_FILL_VEC_D0_ML(candidate, bdtOutputPrompt, mlProbD0, mlProbD0bar, 2);
228+
CHECK_AND_FILL_VEC_D0_ML(candidate, bdtOutputPrompt, mlProbD0, mlProbD0bar, 1);
229+
CHECK_AND_FILL_VEC_D0_ML(candidate, bdtOutputNonPrompt, mlProbD0, mlProbD0bar, 2);
230230

231231
CHECK_AND_FILL_VEC_D0(maxNormalisedDeltaIP);
232232
CHECK_AND_FILL_VEC_D0_FULL(candidate, impactParameterProduct, impactParameterProduct);

PWGHF/D2H/Tasks/taskD0.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -366,8 +366,8 @@ struct HfTaskD0 {
366366

367367
// Insert ML scores after pt (position 2) to match taskDplus structure: [mass, pt, mlScores, ...]
368368
if (doprocessDataWithDCAFitterNMlWithUpc) {
369-
axes.insert(axes.begin() + 2, thnAxisPromptScore);
370369
axes.insert(axes.begin() + 2, thnAxisNonPromptScore);
370+
axes.insert(axes.begin() + 2, thnAxisPromptScore);
371371
axes.insert(axes.begin() + 2, thnAxisBkgScore);
372372
} else {
373373
axes.insert(axes.begin(), thnAxisPromptScore);

0 commit comments

Comments
 (0)