|
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 | + |
17 | 20 | #include "PWGHF/Core/HfMlResponseXic0ToXiPi.h" |
18 | 21 | #include "PWGHF/Core/SelectorCuts.h" |
| 22 | + |
19 | 23 | #include "PWGHF/DataModel/CandidateReconstructionTables.h" |
20 | 24 | #include "PWGHF/DataModel/CandidateSelectionTables.h" |
21 | 25 | #include "PWGHF/Utils/utilsAnalysis.h" |
|
41 | 45 |
|
42 | 46 | #include <cstdint> |
43 | 47 | #include <cstdlib> |
44 | | -#include <string> |
45 | | -#include <vector> |
46 | 48 |
|
47 | 49 | using namespace o2; |
48 | 50 | using namespace o2::aod; |
@@ -142,14 +144,14 @@ struct HfCandidateSelectorToXiPi { |
142 | 144 | 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"}; |
143 | 145 | Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Number of classes in ML model"}; |
144 | 146 | Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{"feature1", "feature2"}, "Names of ML model input features"}; |
145 | | - |
| 147 | + |
146 | 148 | // CCDB configuration |
147 | 149 | Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"}; |
148 | 150 | Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic0ToXipiKf"}, "Paths of models on CCDB"}; |
149 | 151 | 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)"}; |
150 | 152 | Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"}; |
151 | 153 | Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"}; |
152 | | - |
| 154 | + |
153 | 155 | o2::analysis::HfMlResponseXic0ToXiPi<float> hfMlResponse; |
154 | 156 | std::vector<float> outputMlXic0ToXiPi = {}; |
155 | 157 | o2::ccdb::CcdbApi ccdbApi; |
@@ -556,7 +558,7 @@ struct HfCandidateSelectorToXiPi { |
556 | 558 | } else { |
557 | 559 | registry.fill(HIST("hSelMassCharmBaryon"), 0); |
558 | 560 | } |
559 | | - |
| 561 | + |
560 | 562 | // ML selections |
561 | 563 | if (applyMl) { |
562 | 564 | bool isSelectedMlXic0 = false; |
|
0 commit comments