Skip to content

Commit fb81438

Browse files
authored
[PWGHF] Add TPC crossed rows to HfMlResponseHfTracks (#17956)
1 parent 1c8495f commit fb81438

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

PWGHF/Core/HfMlResponseHfTracks.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ enum class InputFeaturesTracks : uint8_t {
7474
itsChi2NCl,
7575
// TPC quality
7676
tpcNClsFound,
77+
tpcNClsCrossedRows,
7778
tpcCrossedRowsOverFindableCls,
7879
tpcChi2NCl,
7980
tpcFractionSharedCls,
@@ -110,6 +111,7 @@ struct HfTrackMlFeatures {
110111
float itsChi2NCl{0.f};
111112
// TPC quality
112113
float tpcNClsFound{0.f};
114+
float tpcNClsCrossedRows{0.f};
113115
float tpcCrossedRowsOverFindableCls{0.f};
114116
float tpcChi2NCl{0.f};
115117
float tpcFractionSharedCls{0.f};
@@ -171,6 +173,7 @@ class HfMlResponseHfTracks : public HfMlResponse<TypeOutputScore>
171173
CHECK_AND_FILL_VEC_HF_TRACK(itsNClsInnerBarrel);
172174
CHECK_AND_FILL_VEC_HF_TRACK(itsChi2NCl);
173175
CHECK_AND_FILL_VEC_HF_TRACK(tpcNClsFound);
176+
CHECK_AND_FILL_VEC_HF_TRACK(tpcNClsCrossedRows);
174177
CHECK_AND_FILL_VEC_HF_TRACK(tpcCrossedRowsOverFindableCls);
175178
CHECK_AND_FILL_VEC_HF_TRACK(tpcChi2NCl);
176179
CHECK_AND_FILL_VEC_HF_TRACK(tpcFractionSharedCls);
@@ -204,6 +207,7 @@ class HfMlResponseHfTracks : public HfMlResponse<TypeOutputScore>
204207
FILL_MAP_HF_TRACK(itsNClsInnerBarrel),
205208
FILL_MAP_HF_TRACK(itsChi2NCl),
206209
FILL_MAP_HF_TRACK(tpcNClsFound),
210+
FILL_MAP_HF_TRACK(tpcNClsCrossedRows),
207211
FILL_MAP_HF_TRACK(tpcCrossedRowsOverFindableCls),
208212
FILL_MAP_HF_TRACK(tpcChi2NCl),
209213
FILL_MAP_HF_TRACK(tpcFractionSharedCls),

PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -525,6 +525,7 @@ struct HfTrackSelectorTagSelTracks {
525525
features.itsNClsInnerBarrel = static_cast<float>(track.itsNClsInnerBarrel());
526526
features.itsChi2NCl = track.itsChi2NCl();
527527
features.tpcNClsFound = static_cast<float>(track.tpcNClsFound());
528+
features.tpcNClsCrossedRows = static_cast<float>(track.tpcNClsCrossedRows());
528529
features.tpcCrossedRowsOverFindableCls = track.tpcCrossedRowsOverFindableCls();
529530
features.tpcChi2NCl = track.tpcChi2NCl();
530531
features.tpcFractionSharedCls = track.tpcFractionSharedCls();

0 commit comments

Comments
 (0)