@@ -69,7 +69,7 @@ struct HfProducerCharmHadronsCharmFemtoDream {
6969 struct HfMlConfig : ConfigurableGroup {
7070 std::string prefix;
7171 static inline const std::array<double , 3 > defaultCuts{1 ., 0 ., 0 .};
72- Configurable<int > applyMlMode{ " applyMlMode " , FillMlFromSelector, " 0: no ML, 1: selector scores, 2: new BDT after selector" };
72+ Configurable<int > mlApplicationMode{ " mlApplicationMode " , FillMlFromSelector, " 0: no ML, 1: selector scores, 2: new BDT after selector" };
7373 Configurable<std::vector<double >> binsPtMl{" binsPtMl" , std::vector<double >{0 ., 36 .}, " pT bin limits for new BDT" };
7474 Configurable<LabeledArray<double >> cutsMl{" cutsMl" , {defaultCuts.data (), 1 , 3 }, " New BDT cuts per pT bin: background, prompt, nonprompt" };
7575 Configurable<std::vector<int >> cutDirMl{" cutDirMl" , std::vector<int >{0 , 1 , 1 }, " Reject scores above (0), below (1), or do not cut (2)" };
@@ -118,10 +118,10 @@ struct HfProducerCharmHadronsCharmFemtoDream {
118118 template <typename Response>
119119 void initMl (HfMlConfig const & cfg, Response& response, bool withMl)
120120 {
121- if (cfg.applyMlMode .value < NoMl || cfg.applyMlMode .value > FillMlFromNewBDT) {
122- LOGP (fatal, " {}: invalid applyMlMode " , cfg.prefix );
121+ if (cfg.mlApplicationMode .value < NoMl || cfg.mlApplicationMode .value > FillMlFromNewBDT) {
122+ LOGP (fatal, " {}: invalid mlApplicationMode " , cfg.prefix );
123123 }
124- if (cfg.applyMlMode .value != FillMlFromNewBDT) {
124+ if (cfg.mlApplicationMode .value != FillMlFromNewBDT) {
125125 return ;
126126 }
127127 if (!withMl) {
@@ -238,10 +238,10 @@ struct HfProducerCharmHadronsCharmFemtoDream {
238238 }
239239 std::array<float , 3 > scores{-1 .f , -1 .f , -1 .f };
240240 if constexpr (WithMl) {
241- if (mlD0.applyMlMode != NoMl) {
241+ if (mlD0.mlApplicationMode != NoMl) {
242242 scores = hypothesis == 0 ? readScores (cand.mlProbD0 ()) : readScores (cand.mlProbD0bar ());
243243 }
244- if (mlD0.applyMlMode == FillMlFromNewBDT) {
244+ if (mlD0.mlApplicationMode == FillMlFromNewBDT) {
245245 // Do not call the ML response with an out-of-range model index.
246246 if (!std::isfinite (cand.pt ()) || cand.pt () < mlD0.binsPtMl .value .front () || cand.pt () >= mlD0.binsPtMl .value .back ()) {
247247 continue ;
@@ -272,10 +272,10 @@ struct HfProducerCharmHadronsCharmFemtoDream {
272272 auto soft = cand.template prongPi_as <aod::Tracks>();
273273 std::array<float , 3 > scores{-1 .f , -1 .f , -1 .f };
274274 if constexpr (WithMl) {
275- if (mlDstar.applyMlMode != NoMl) {
275+ if (mlDstar.mlApplicationMode != NoMl) {
276276 scores = readScores (cand.mlProbDstarToD0Pi ());
277277 }
278- if (mlDstar.applyMlMode == FillMlFromNewBDT) {
278+ if (mlDstar.mlApplicationMode == FillMlFromNewBDT) {
279279 if (!std::isfinite (cand.pt ()) || cand.pt () < mlDstar.binsPtMl .value .front () || cand.pt () >= mlDstar.binsPtMl .value .back ()) {
280280 continue ;
281281 }
0 commit comments