PWGEM/Dilepton: update MlResponse

Author: dsekihat

PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h

@@ -144,6 +144,7 @@ enum class InputFeaturesDielectronSingleTrack : uint8_t {
   tgl,
   isAssociatedToMPC,
   tpcNClsFound,
+  tpcNClsPID,
   tpcNClsCrossedRows,
   tpcCrossedRowsOverFindableCls,
   tpcFoundOverFindableCls,
@@ -266,6 +267,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tgl);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsFound);
+      CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsPID);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls);
@@ -414,6 +416,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tgl),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsFound),
+      FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsPID),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls),

PWGEM/Dilepton/Utils/MlResponseO2Track.h

@@ -160,6 +160,7 @@ enum class InputFeaturesO2Track : uint8_t {
   tpctofNSigmaKa,
   tpctofNSigmaPr,
   tpcNClsFound,
+  tpcNClsPID,
   tpcNClsCrossedRows,
   tpcChi2NCl,
   hasITS,
@@ -215,6 +216,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
       CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaKa, tpcNSigmaKa, tofNSigmaKa, hasTOF);
       CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaPr, tpcNSigmaPr, tofNSigmaPr, hasTOF);
       CHECK_AND_FILL_O2_TRACK(tpcNClsFound);
+      CHECK_AND_FILL_O2_TRACK(tpcNClsPID);
       CHECK_AND_FILL_O2_TRACK(tpcNClsCrossedRows);
       CHECK_AND_FILL_O2_TRACK(tpcChi2NCl);
       CHECK_AND_FILL_O2_TRACK(hasITS);
@@ -296,6 +298,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
       FILL_MAP_O2_TRACK(tpctofNSigmaKa),
       FILL_MAP_O2_TRACK(tpctofNSigmaPr),
       FILL_MAP_O2_TRACK(tpcNClsFound),
+      FILL_MAP_O2_TRACK(tpcNClsPID),
       FILL_MAP_O2_TRACK(tpcNClsCrossedRows),
       FILL_MAP_O2_TRACK(tpcChi2NCl),
       FILL_MAP_O2_TRACK(hasITS),