Skip to content

Commit 7fb2719

Browse files
dsekihatdsekihat
authored
[PWGEM/Dilepton] fix for thresholds in PID ML (#11705)
1 parent 13b446a commit 7fb2719

File tree

6 files changed

+52
-51
lines changed

6 files changed

+52
-51
lines changed

‎PWGEM/Dilepton/Core/Dilepton.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -707,14 +707,14 @@ struct Dilepton {
707707

708708
if (dielectroncuts.cfg_pid_scheme == static_cast<int>(DielectronCut::PIDSchemes::kPIDML)) { // please call this at the end of DefineDileptonCut
709709
static constexpr int nClassesMl = 2;
710-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
711-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
710+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
711+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
712712
const uint32_t nBinsMl = dielectroncuts.binsMl.value.size() - 1;
713713
const std::vector<std::string> labelsBins(nBinsMl, "bin");
714714
double cutsMlArr[nBinsMl][nClassesMl];
715715
for (uint32_t i = 0; i < nBinsMl; i++) {
716-
cutsMlArr[i][0] = dielectroncuts.cutsMl.value[i];
717-
cutsMlArr[i][1] = 0.;
716+
cutsMlArr[i][0] = 0.;
717+
cutsMlArr[i][1] = dielectroncuts.cutsMl.value[i];
718718
}
719719
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
720720

‎PWGEM/Dilepton/Core/DileptonMC.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -672,14 +672,14 @@ struct DileptonMC {
672672

673673
if (dielectroncuts.cfg_pid_scheme == static_cast<int>(DielectronCut::PIDSchemes::kPIDML)) { // please call this at the end of DefineDileptonCut
674674
static constexpr int nClassesMl = 2;
675-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
676-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
675+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
676+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
677677
const uint32_t nBinsMl = dielectroncuts.binsMl.value.size() - 1;
678678
const std::vector<std::string> labelsBins(nBinsMl, "bin");
679679
double cutsMlArr[nBinsMl][nClassesMl];
680680
for (uint32_t i = 0; i < nBinsMl; i++) {
681-
cutsMlArr[i][0] = dielectroncuts.cutsMl.value[i];
682-
cutsMlArr[i][1] = 0.;
681+
cutsMlArr[i][0] = 0.;
682+
cutsMlArr[i][1] = dielectroncuts.cutsMl.value[i];
683683
}
684684
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
685685

‎PWGEM/Dilepton/Core/PhotonHBT.h‎

Lines changed: 32 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -17,40 +17,41 @@
1717
#ifndef PWGEM_DILEPTON_CORE_PHOTONHBT_H_
1818
#define PWGEM_DILEPTON_CORE_PHOTONHBT_H_
1919

20+
#include "PWGEM/Dilepton/Core/DielectronCut.h"
21+
#include "PWGEM/Dilepton/Core/EMEventCut.h"
22+
#include "PWGEM/Dilepton/Utils/EMTrack.h"
23+
#include "PWGEM/Dilepton/Utils/EMTrackUtilities.h"
24+
#include "PWGEM/Dilepton/Utils/EventHistograms.h"
25+
#include "PWGEM/Dilepton/Utils/EventMixingHandler.h"
26+
#include "PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h"
27+
#include "PWGEM/PhotonMeson/Core/V0PhotonCut.h"
28+
#include "PWGEM/PhotonMeson/DataModel/gammaTables.h"
29+
30+
#include "Tools/ML/MlResponse.h"
31+
32+
#include "CCDB/BasicCCDBManager.h"
33+
#include "DataFormatsParameters/GRPMagField.h"
34+
#include "DataFormatsParameters/GRPObject.h"
35+
#include "DetectorsBase/GeometryManager.h"
36+
#include "Framework/ASoAHelpers.h"
37+
#include "Framework/AnalysisDataModel.h"
38+
#include "Framework/AnalysisTask.h"
39+
#include "Framework/runDataProcessing.h"
40+
41+
#include "Math/GenVector/Boost.h"
42+
#include "Math/Vector3D.h"
43+
#include "Math/Vector4D.h"
44+
#include "TString.h"
45+
2046
#include <algorithm>
2147
#include <cstring>
2248
#include <iterator>
2349
#include <map>
50+
#include <random>
2451
#include <string>
25-
#include <utility>
2652
#include <tuple>
53+
#include <utility>
2754
#include <vector>
28-
#include <random>
29-
30-
#include "TString.h"
31-
#include "Math/Vector4D.h"
32-
#include "Math/Vector3D.h"
33-
#include "Math/GenVector/Boost.h"
34-
#include "Framework/runDataProcessing.h"
35-
#include "Framework/AnalysisTask.h"
36-
#include "Framework/AnalysisDataModel.h"
37-
#include "Framework/ASoAHelpers.h"
38-
39-
#include "DetectorsBase/GeometryManager.h"
40-
#include "DataFormatsParameters/GRPObject.h"
41-
#include "DataFormatsParameters/GRPMagField.h"
42-
#include "CCDB/BasicCCDBManager.h"
43-
#include "Tools/ML/MlResponse.h"
44-
45-
#include "PWGEM/Dilepton/Utils/EMTrackUtilities.h"
46-
#include "PWGEM/Dilepton/Core/EMEventCut.h"
47-
#include "PWGEM/Dilepton/Core/DielectronCut.h"
48-
#include "PWGEM/Dilepton/Utils/EMTrack.h"
49-
#include "PWGEM/Dilepton/Utils/EventMixingHandler.h"
50-
#include "PWGEM/Dilepton/Utils/EventHistograms.h"
51-
#include "PWGEM/PhotonMeson/DataModel/gammaTables.h"
52-
#include "PWGEM/PhotonMeson/Core/V0PhotonCut.h"
53-
#include "PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h"
5455

5556
namespace o2::aod::pwgem::dilepton::core::photonhbt
5657
{
@@ -589,14 +590,14 @@ struct PhotonHBT {
589590

590591
if (dielectroncuts.cfg_pid_scheme == static_cast<int>(DielectronCut::PIDSchemes::kPIDML)) { // please call this at the end of DefineDileptonCut
591592
static constexpr int nClassesMl = 2;
592-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
593-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
593+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
594+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
594595
const uint32_t nBinsMl = dielectroncuts.binsMl.value.size() - 1;
595596
const std::vector<std::string> labelsBins(nBinsMl, "bin");
596597
double cutsMlArr[nBinsMl][nClassesMl];
597598
for (uint32_t i = 0; i < nBinsMl; i++) {
598-
cutsMlArr[i][0] = dielectroncuts.cutsMl.value[i];
599-
cutsMlArr[i][1] = 0.;
599+
cutsMlArr[i][0] = 0.;
600+
cutsMlArr[i][1] = dielectroncuts.cutsMl.value[i];
600601
}
601602
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
602603

‎PWGEM/Dilepton/Core/SingleTrackQC.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -415,14 +415,14 @@ struct SingleTrackQC {
415415

416416
if (dielectroncuts.cfg_pid_scheme == static_cast<int>(DielectronCut::PIDSchemes::kPIDML)) { // please call this at the end of DefineDileptonCut
417417
static constexpr int nClassesMl = 2;
418-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
419-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
418+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
419+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
420420
const uint32_t nBinsMl = dielectroncuts.binsMl.value.size() - 1;
421421
const std::vector<std::string> labelsBins(nBinsMl, "bin");
422422
double cutsMlArr[nBinsMl][nClassesMl];
423423
for (uint32_t i = 0; i < nBinsMl; i++) {
424-
cutsMlArr[i][0] = dielectroncuts.cutsMl.value[i];
425-
cutsMlArr[i][1] = 0.;
424+
cutsMlArr[i][0] = 0.;
425+
cutsMlArr[i][1] = dielectroncuts.cutsMl.value[i];
426426
}
427427
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
428428

‎PWGEM/Dilepton/Core/SingleTrackQCMC.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -474,14 +474,14 @@ struct SingleTrackQCMC {
474474

475475
if (dielectroncuts.cfg_pid_scheme == static_cast<int>(DielectronCut::PIDSchemes::kPIDML)) { // please call this at the end of DefineDileptonCut
476476
static constexpr int nClassesMl = 2;
477-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
478-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
477+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
478+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
479479
const uint32_t nBinsMl = dielectroncuts.binsMl.value.size() - 1;
480480
const std::vector<std::string> labelsBins(nBinsMl, "bin");
481481
double cutsMlArr[nBinsMl][nClassesMl];
482482
for (uint32_t i = 0; i < nBinsMl; i++) {
483-
cutsMlArr[i][0] = dielectroncuts.cutsMl.value[i];
484-
cutsMlArr[i][1] = 0.;
483+
cutsMlArr[i][0] = 0.;
484+
cutsMlArr[i][1] = dielectroncuts.cutsMl.value[i];
485485
}
486486
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
487487

‎PWGEM/Dilepton/TableProducer/skimmerPrimaryElectron.cxx‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -182,14 +182,14 @@ struct skimmerPrimaryElectron {
182182

183183
if (usePIDML) {
184184
static constexpr int nClassesMl = 2;
185-
const std::vector<int> cutDirMl = {o2::cuts_ml::CutGreater, o2::cuts_ml::CutNot};
186-
const std::vector<std::string> labelsClasses = {"Signal", "Background"};
185+
const std::vector<int> cutDirMl = {o2::cuts_ml::CutNot, o2::cuts_ml::CutSmaller};
186+
const std::vector<std::string> labelsClasses = {"Background", "Signal"};
187187
const uint32_t nBinsMl = binsMl.value.size() - 1;
188188
const std::vector<std::string> labelsBins(nBinsMl, "bin");
189189
double cutsMlArr[nBinsMl][nClassesMl];
190190
for (uint32_t i = 0; i < nBinsMl; i++) {
191-
cutsMlArr[i][0] = cutsMl.value[i];
192-
cutsMlArr[i][1] = 0.;
191+
cutsMlArr[i][0] = 0.0;
192+
cutsMlArr[i][1] = cutsMl.value[i];
193193
}
194194
o2::framework::LabeledArray<double> cutsMl = {cutsMlArr[0], nBinsMl, nClassesMl, labelsBins, labelsClasses};
195195

0 commit comments

Comments
 (0)