Skip to content

Commit e95cd43

Browse files
gianniliveraroddobrigkalibuild
authored
PWGLF: add histo to check candidate selection (#5642)
* ML and Sigma0 work * PWGLF: Updating feature list and selections * PWGLF/Sigma0builder: modularize further * Lambdakzerobuilder tuning * Builder incorporates ML * Move ML calculation to cover both V0 and V0fC * PWGLF/ML: Changing task name, minor adjustments in ML selection * Eval only if requested * implementation of ML in cascade builder * Add cascade ML selection TC * PWGLF/ML: Including ITSClSizes in feature list * Forgot git add * Modifications for cascades * Add mass window conformity map * PWGLF/ML: Major adjustment in TC * Add mass windows * Changes in pre-selections * Open defaults * Open defaults * Please consider the following formatting changes * First batch of megalinter errors * Please consider the following formatting changes * Megalinter order errors fixed * Please consider the following formatting changes * Fix one last ordering issue * Fix LAST megalinter error phew phew * Including histogram to save selection info * Changing selection criteria * Please consider the following formatting changes --------- Co-authored-by: ddobrigk <david.dobrigkeit.chinellato@cern.ch> Co-authored-by: ALICE Action Bot <alibuild@cern.ch>
1 parent 644dee9 commit e95cd43

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

PWGLF/TableProducer/Strangeness/lambdakzeroMLSelectionTreeCreator.cxx

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -107,13 +107,12 @@ struct lambdakzeroMLSelectionTreeCreator {
107107

108108
// Axis:
109109
ConfigurableAxis centralityAxis{"centralityAxis", {100, 0.0f, 100.0f}, ""};
110-
111-
// Save certain candidates only
112-
Configurable<int> saveCandidateMask{"saveCandidateMask", 15, "Mask to decide saving candidates"};
110+
ConfigurableAxis candSelectionAxis{"candSelectionAxis", {16, 0.0f, 16.0f}, ""};
113111

114112
void init(InitContext const&)
115113
{
116114
histos.add("hEventCentrality", "hEventCentrality", kTH1F, {centralityAxis});
115+
histos.add("hCandSelection", "hCandSelection", kTH1F, {candSelectionAxis});
117116
}
118117

119118
// Helper struct to pass v0 information
@@ -245,7 +244,9 @@ struct lambdakzeroMLSelectionTreeCreator {
245244
// 13: Consistent with Lambda, Gamma, and K0Short, 14: Consistent with Anti-Lambda, Gamma, and K0Short
246245
// 15: Consistent with Lambda, Anti-Lambda, Gamma, and K0Short
247246

248-
if ((Candidate.SelHypothesis & saveCandidateMask) != saveCandidateMask)
247+
histos.fill(HIST("hCandSelection"), Candidate.SelHypothesis);
248+
249+
if (Candidate.SelHypothesis == 0)
249250
return;
250251

251252
// MC flags

PWGLF/TableProducer/Strangeness/lambdakzeromlselection.cxx

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ using V0DerivedDatas = soa::Join<aod::V0Cores, aod::V0Extras, aod::V0CollRefs>;
6969

7070
struct lambdakzeromlselection {
7171
o2::ml::OnnxModel lambda_bdt;
72-
;
7372
o2::ml::OnnxModel antilambda_bdt;
7473
o2::ml::OnnxModel gamma_bdt;
7574
o2::ml::OnnxModel kzeroshort_bdt;
@@ -188,14 +187,14 @@ struct lambdakzeromlselection {
188187
void processDerivedData(aod::StraCollision const& coll, V0DerivedDatas const& v0s)
189188
{
190189
histos.fill(HIST("hEventVertexZ"), coll.posZ());
191-
for (auto& v0 : v0s) { // looping over lambdas
190+
for (auto& v0 : v0s) {
192191
processCandidate(v0);
193192
}
194193
}
195194
void processStandardData(aod::Collision const& coll, V0OriginalDatas const& v0s)
196195
{
197196
histos.fill(HIST("hEventVertexZ"), coll.posZ());
198-
for (auto& v0 : v0s) { // looping over lambdas
197+
for (auto& v0 : v0s) {
199198
processCandidate(v0);
200199
}
201200
}

0 commit comments

Comments
 (0)