@@ -1695,9 +1695,6 @@ struct HfTrackIndexSkimCreator {
16951695 const std::vector<double > ptBinsMl{0 ., 1 .e10 };
16961696 const std::vector<int > cutDirMl{o2::cuts_ml::CutDirection::CutGreater, o2::cuts_ml::CutDirection::CutSmaller, o2::cuts_ml::CutDirection::CutSmaller};
16971697 const std::array<LabeledArray<double >, kN3ProngDecaysUsedMlForHfFilters > thresholdMlScore3Prongs{config.thresholdMlScoreDplusToPiKPi , config.thresholdMlScoreLcToPiKP , config.thresholdMlScoreDsToPiKK , config.thresholdMlScoreXicToPiKP };
1698- const std::vector<std::string> inputFeatures2Prongs = {" ptProng0" , " dcaXyProng0" , " dcaZProng0" , " ptProng1" , " dcaXyProng1" , " dcaZProng1" };
1699- const std::vector<std::string> inputFeatures3Prongs = {" ptProng0" , " dcaXyProng0" , " dcaZProng0" , " ptProng1" , " dcaXyProng1" , " dcaZProng1" , " ptProng2" , " dcaXyProng2" , " dcaZProng2" };
1700- const std::vector<std::string> inputFeatures3ProngsWithPid = {" ptProng0" , " dcaXyProng0" , " dcaZProng0" , " ptProng1" , " dcaXyProng1" , " dcaZProng1" , " ptProng2" , " dcaXyProng2" , " dcaZProng2" , " tpcNSigmaPrProng0" , " tpcNSigmaPrProng2" , " tpcNSigmaPiProng0" , " tpcNSigmaPiProng2" , " tpcNSigmaKaProng1" };
17011698
17021699 // initialise 2-prong ML response
17031700 hfMlResponse2Prongs.configure (ptBinsMl, config.thresholdMlScoreD0ToKPi , cutDirMl, 3 );
@@ -1707,7 +1704,6 @@ struct HfTrackIndexSkimCreator {
17071704 } else {
17081705 hfMlResponse2Prongs.setModelPathsLocal (onnxFileNames2Prongs);
17091706 }
1710- hfMlResponse2Prongs.cacheInputFeaturesIndices (inputFeatures2Prongs);
17111707 hfMlResponse2Prongs.init ();
17121708
17131709 // initialise 3-prong ML responses
@@ -1723,11 +1719,6 @@ struct HfTrackIndexSkimCreator {
17231719 } else {
17241720 hfMlResponse3Prongs[iDecay3P].setModelPathsLocal (onnxFileNames3Prongs[iDecay3P]);
17251721 }
1726- if ((doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt || doprocess2And3ProngsNoPvRefitWithPidForHfFiltersBdt) && iDecay3P == aod::hf_cand_3prong::DecayType::LcToPKPi) {
1727- hfMlResponse3Prongs[iDecay3P].cacheInputFeaturesIndices (inputFeatures3ProngsWithPid);
1728- } else {
1729- hfMlResponse3Prongs[iDecay3P].cacheInputFeaturesIndices (inputFeatures3Prongs);
1730- }
17311722 hfMlResponse3Prongs[iDecay3P].init ();
17321723 }
17331724 }
0 commit comments