diff --git a/PWGDQ/Tasks/qaMatching.cxx b/PWGDQ/Tasks/qaMatching.cxx index 9455bbc86c7..bf0e7beb53a 100644 --- a/PWGDQ/Tasks/qaMatching.cxx +++ b/PWGDQ/Tasks/qaMatching.cxx @@ -362,12 +362,12 @@ struct QaMatching { std::array*, MlModelsNum> modelNames{ &cfgMlModelName1, &cfgMlModelName2, &cfgMlModelName3, &cfgMlModelName4, &cfgMlModelName5}; - Configurable cfgMlInputFeatures1{"cfgMlInputFeatures1", "chi2MCHMFT", "Names of ML model input features"}; - Configurable cfgMlInputFeatures2{"cfgMlInputFeatures2", std::string{""}, "Names of ML model input features"}; - Configurable cfgMlInputFeatures3{"cfgMlInputFeatures3", std::string{""}, "Names of ML model input features"}; - Configurable cfgMlInputFeatures4{"cfgMlInputFeatures4", std::string{""}, "Names of ML model input features"}; - Configurable cfgMlInputFeatures5{"cfgMlInputFeatures5", std::string{""}, "Names of ML model input features"}; - std::array*, MlModelsNum> inputFeatures{ + Configurable> cfgMlInputFeatures1{"cfgMlInputFeatures1", std::vector{"chi2MCHMFT"}, "Names of ML model input features"}; + Configurable> cfgMlInputFeatures2{"cfgMlInputFeatures2", std::vector{}, "Names of ML model input features"}; + Configurable> cfgMlInputFeatures3{"cfgMlInputFeatures3", std::vector{}, "Names of ML model input features"}; + Configurable> cfgMlInputFeatures4{"cfgMlInputFeatures4", std::vector{}, "Names of ML model input features"}; + Configurable> cfgMlInputFeatures5{"cfgMlInputFeatures5", std::vector{}, "Names of ML model input features"}; + std::array>*, MlModelsNum> inputFeatures{ &cfgMlInputFeatures1, &cfgMlInputFeatures2, &cfgMlInputFeatures3, &cfgMlInputFeatures4, &cfgMlInputFeatures5}; Configurable cfgMlModelMatchingScoreCut1{"cfgMlModelMatchingScoreCut1", 0.f, "Minimum score value for selecting good matches"}; @@ -1158,12 +1158,12 @@ struct QaMatching { auto matchingPlaneZ = configMlOptions.matchingPlaneZs[modelId]->value; auto extrapMethod = configMlOptions.matchingExtrapMethods[modelId]->value; - if (label == "" || modelPath == "" || inputFeatures == "" || modelName == "") + if (label == "" || modelPath == "" || inputFeatures.empty() || modelName == "") break; matchingMlResponses[label].configure(binsPtMl, mycutsMl, cutDirMl, 1); matchingMlResponses[label].setModelPathsCCDB(std::vector{modelName}, fCCDBApi, std::vector{modelPath}, configCcdb.cfgCcdbNoLaterThan.value); - matchingMlResponses[label].cacheInputFeaturesIndices(std::vector{inputFeatures}); + matchingMlResponses[label].cacheInputFeaturesIndices(inputFeatures); matchingMlResponses[label].init(); matchingScoreCuts[label] = scoreMin;