Skip to content

Commit b23b684

Browse files
authored
[PWGEM/Dilepton] update MlResponse (#13190)
1 parent 44edb11 commit b23b684

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ enum class InputFeaturesDielectronSingleTrack : uint8_t {
144144
tgl,
145145
isAssociatedToMPC,
146146
tpcNClsFound,
147+
tpcNClsPID,
147148
tpcNClsCrossedRows,
148149
tpcCrossedRowsOverFindableCls,
149150
tpcFoundOverFindableCls,
@@ -266,6 +267,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
266267
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tgl);
267268
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC);
268269
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsFound);
270+
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsPID);
269271
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows);
270272
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls);
271273
CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls);
@@ -414,6 +416,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
414416
FILL_MAP_DIELECTRON_SINGLE_TRACK(tgl),
415417
FILL_MAP_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC),
416418
FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsFound),
419+
FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsPID),
417420
FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows),
418421
FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls),
419422
FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls),

PWGEM/Dilepton/Utils/MlResponseO2Track.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@ enum class InputFeaturesO2Track : uint8_t {
160160
tpctofNSigmaKa,
161161
tpctofNSigmaPr,
162162
tpcNClsFound,
163+
tpcNClsPID,
163164
tpcNClsCrossedRows,
164165
tpcChi2NCl,
165166
hasITS,
@@ -215,6 +216,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
215216
CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaKa, tpcNSigmaKa, tofNSigmaKa, hasTOF);
216217
CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaPr, tpcNSigmaPr, tofNSigmaPr, hasTOF);
217218
CHECK_AND_FILL_O2_TRACK(tpcNClsFound);
219+
CHECK_AND_FILL_O2_TRACK(tpcNClsPID);
218220
CHECK_AND_FILL_O2_TRACK(tpcNClsCrossedRows);
219221
CHECK_AND_FILL_O2_TRACK(tpcChi2NCl);
220222
CHECK_AND_FILL_O2_TRACK(hasITS);
@@ -296,6 +298,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
296298
FILL_MAP_O2_TRACK(tpctofNSigmaKa),
297299
FILL_MAP_O2_TRACK(tpctofNSigmaPr),
298300
FILL_MAP_O2_TRACK(tpcNClsFound),
301+
FILL_MAP_O2_TRACK(tpcNClsPID),
299302
FILL_MAP_O2_TRACK(tpcNClsCrossedRows),
300303
FILL_MAP_O2_TRACK(tpcChi2NCl),
301304
FILL_MAP_O2_TRACK(hasITS),

0 commit comments

Comments
 (0)