You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: PWGEM/Dilepton/TableProducer/filterEoI.cxx
+4-1Lines changed: 4 additions & 1 deletion
Original file line number
Diff line number
Diff line change
@@ -31,6 +31,7 @@
31
31
#include<TH1.h>
32
32
33
33
#include<cstdint>
34
+
#include<string>
34
35
35
36
usingnamespaceo2;
36
37
usingnamespaceo2::framework;
@@ -49,12 +50,14 @@ struct filterEoI {
49
50
Configurable<bool> inheritFromOtherTask{"inheritFromOtherTask", true, "Flag to iherit all common configurables from skimmerPrimaryElectron or skimmerPrimaryMuon"};
50
51
Configurable<int> minNelectron{"minNelectron", -1, "min number of electron candidates per collision"};
51
52
Configurable<int> minNmuon{"minNmuon", -1, "min number of muon candidates per collision"};
53
+
Configurable<std::string> taskNameForNelectron{"taskNameForNelectron", "skimmer-primary-electron", "task name where minNelectron is defined."};
54
+
Configurable<std::string> varNameForNelectron{"varNameForNelectron", "minNelectron", "variable name for minNelectron"};
52
55
53
56
HistogramRegistry fRegistry{"output"};
54
57
voidinit(o2::framework::InitContext& initContext)
55
58
{
56
59
if (inheritFromOtherTask.value) { // Inheriting from other task
o2::framework::Configurable<bool> usePIDML{"usePIDML", false, "Flag to use PID ML"};
124
124
o2::framework::Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"filename"}, "ONNX file names for each bin (if not from CCDB full path)"};
125
125
o2::framework::Configurable<std::vector<std::string>> onnxPathsCCDB{"onnxPathsCCDB", std::vector<std::string>{"path"}, "Paths of models on CCDB"};
126
-
o2::framework::Configurable<std::vector<double>> binsMl{"binsMl", std::vector<double>{0.1, 0.15, 0.2, 0.25, 0.4, 0.8, 1.6, 2.0, 20}, "Bin limits for ML application"};
o2::framework::Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{"tpcInnerParam", "tpcNClsFound", "tpcChi2NCl", "tpcNSigmaEl", "tofNSigmaEl", "meanClusterSizeITSobCosTgl"}, "Names of ML model input features"};
129
129
o2::framework::Configurable<std::string> nameBinningFeature{"nameBinningFeature", "tpcInnerParam", "Names of ML model binning feature"};
130
130
o2::framework::Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"};
0 commit comments