Skip to content

Commit c4ae5a5

Browse files
Rename applyMlMode to mlApplicationMode
1 parent ab6e7dd commit c4ae5a5

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

‎PWGHF/HFC/TableProducer/producerCharmHadronsCharmFemtoDream.cxx‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)