|
14 | 14 | /// \author Federica Zanone <federica.zanone@cern.ch>, Heidelberg University |
15 | 15 | /// \author Tao Fang <tao.fang@cern.ch>, Central China Normal University |
16 | 16 |
|
17 | | -#include <string> |
18 | | -#include <vector> |
19 | | - |
20 | 17 | #include "PWGHF/Core/HfMlResponseXic0ToXiPi.h" |
21 | 18 | #include "PWGHF/Core/SelectorCuts.h" |
22 | | - |
23 | 19 | #include "PWGHF/DataModel/CandidateReconstructionTables.h" |
24 | 20 | #include "PWGHF/DataModel/CandidateSelectionTables.h" |
25 | 21 | #include "PWGHF/Utils/utilsAnalysis.h" |
|
45 | 41 |
|
46 | 42 | #include <cstdint> |
47 | 43 | #include <cstdlib> |
| 44 | +#include <string> |
| 45 | +#include <vector> |
48 | 46 |
|
49 | 47 | using namespace o2; |
50 | 48 | using namespace o2::aod; |
@@ -144,14 +142,14 @@ struct HfCandidateSelectorToXiPi { |
144 | 142 | Configurable<LabeledArray<double>> cutsMl{"cutsMl", {hf_cuts_ml::Cuts[0], hf_cuts_ml::NBinsPt, hf_cuts_ml::NCutScores, hf_cuts_ml::labelsPt, hf_cuts_ml::labelsCutScore}, "ML selections per pT bin"}; |
145 | 143 | Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Number of classes in ML model"}; |
146 | 144 | Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{"feature1", "feature2"}, "Names of ML model input features"}; |
147 | | - |
| 145 | + |
148 | 146 | // CCDB configuration |
149 | 147 | Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"}; |
150 | 148 | Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic0ToXipiKf"}, "Paths of models on CCDB"}; |
151 | 149 | Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_Xic0ToXipiKf.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"}; |
152 | 150 | Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"}; |
153 | 151 | Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"}; |
154 | | - |
| 152 | + |
155 | 153 | o2::analysis::HfMlResponseXic0ToXiPi<float> hfMlResponse; |
156 | 154 | std::vector<float> outputMlXic0ToXiPi = {}; |
157 | 155 | o2::ccdb::CcdbApi ccdbApi; |
@@ -558,7 +556,7 @@ struct HfCandidateSelectorToXiPi { |
558 | 556 | } else { |
559 | 557 | registry.fill(HIST("hSelMassCharmBaryon"), 0); |
560 | 558 | } |
561 | | - |
| 559 | + |
562 | 560 | // ML selections |
563 | 561 | if (applyMl) { |
564 | 562 | bool isSelectedMlXic0 = false; |
|
0 commit comments