@@ -427,7 +427,7 @@ class ElectronModule
427427
428428 if (fConfigDFeT .useML ) {
429429 static constexpr int nClassesMl = 4 ;
430- const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller };
430+ const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot };
431431 const std::vector<std::string> labelsClasses = {" prompt" , " prompthc" , " nonprompthc" , " hb" };
432432 const uint32_t nBinsMl = fConfigDFeT .binsMl .value .size () - 1 ;
433433 const std::vector<std::string> labelsBins (nBinsMl, " bin" );
@@ -453,7 +453,7 @@ class ElectronModule
453453
454454 if (fConfigDFeV0 .useML ) {
455455 static constexpr int nClassesMl = 4 ;
456- const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller };
456+ const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot };
457457 const std::vector<std::string> labelsClasses = {" prompt" , " prompthc" , " nonprompthc" , " hb" };
458458 const uint32_t nBinsMl = fConfigDFeV0 .binsMl .value .size () - 1 ;
459459 const std::vector<std::string> labelsBins (nBinsMl, " bin" );
@@ -479,7 +479,7 @@ class ElectronModule
479479
480480 if (fConfigDFeC .useML ) {
481481 static constexpr int nClassesMl = 4 ;
482- const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller };
482+ const std::vector<int > cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot, o2::cuts_ml::CutNot };
483483 const std::vector<std::string> labelsClasses = {" prompt" , " prompthc" , " nonprompthc" , " hb" };
484484 const uint32_t nBinsMl = fConfigDFeC .binsMl .value .size () - 1 ;
485485 const std::vector<std::string> labelsBins (nBinsMl, " bin" );
0 commit comments