|
15 | 15 | /// |
16 | 16 | /// \author Fabio Catalano <fabio.catalano@cern.ch>, CERN |
17 | 17 |
|
| 18 | +#include "PWGHF/Core/HfHelper.h" |
| 19 | +#include "PWGHF/DataModel/CandidateReconstructionTables.h" |
| 20 | + |
| 21 | +#include "Tools/ML/MlResponse.h" |
| 22 | + |
18 | 23 | #include "Framework/AnalysisTask.h" |
19 | 24 | #include "Framework/HistogramRegistry.h" |
20 | 25 | #include "Framework/runDataProcessing.h" |
21 | 26 |
|
22 | | -#include "Tools/ML/MlResponse.h" |
23 | | -#include "PWGHF/Core/HfHelper.h" |
24 | | -#include "PWGHF/DataModel/CandidateReconstructionTables.h" |
25 | | - |
26 | 27 | using namespace o2; |
27 | 28 | using namespace o2::analysis; |
28 | 29 | using namespace o2::framework; |
@@ -113,26 +114,12 @@ struct applyMlSelection { |
113 | 114 | // Retrieve model output and selection outcome |
114 | 115 |
|
115 | 116 | // Fill BDT score histograms before selection |
116 | | - registry.fill(HIST("hPromptScoreBeforeSel"), outputMl[0]); |
117 | 117 |
|
118 | 118 | // Fill histograms for selected candidates |
119 | | - bool isSelectedMlPiKK = true; |
120 | | - if (isSelectedMlPiKK) { |
121 | | - registry.fill(HIST("hMassAfterSelVsPt"), hfHelper.invMassDsToPiKK(candidate), candidate.pt()); |
122 | | - registry.fill(HIST("hPromptScoreAfterSelVsPt"), outputMl[0], candidate.pt()); |
123 | | - } |
124 | 119 |
|
125 | 120 | outputMl.clear(); // not necessary in this case but for good measure |
126 | 121 |
|
127 | 122 | // Perform ML selections for other mass hypothesis (Ds -> PhiPi -> KKPi) |
128 | | - std::vector<float> inputFeaturesKKPi{candidate.cpa(), |
129 | | - candidate.cpaXY(), |
130 | | - candidate.decayLength(), |
131 | | - candidate.decayLengthXY(), |
132 | | - static_cast<float>(hfHelper.deltaMassPhiDsToKKPi(candidate)), |
133 | | - candidate.impactParameterXY(), |
134 | | - static_cast<float>(hfHelper.cos3PiKDsToKKPi(candidate)), |
135 | | - candidate.maxNormalisedDeltaIP()}; |
136 | 123 |
|
137 | 124 | // Retrieve model output and selection outcome |
138 | 125 |
|
|
0 commit comments