Skip to content

Commit 6a4c788

Browse files
committed
Refine Omegac QA ML implementation
1 parent a793db1 commit 6a4c788

3 files changed

Lines changed: 51 additions & 173 deletions

File tree

‎PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,8 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
104104
/// Method to get the input features vector needed for ML inference
105105
/// \param candidate is the OMEGAC0 candidate
106106
/// \param lamProngPi is the candidate's lamProngPi
107+
/// \param cascProng is the bachelor track from the Omega decay
108+
/// \param charmBaryonProng is the bachelor pion from the OmegaC decay
107109
/// \return inputFeatures vector
108110
template <typename T1, typename T2, typename T3>
109111
std::vector<float> getInputFeatures(T1 const& candidate, T2 const& lamProngPi, T2 const& cascProng, T3 const& charmBaryonProng)

‎PWGHF/D2H/Tasks/taskOmegac0ToOmegaPiQa.cxx‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ struct HfTaskOmegac0ToOmegaPiQa {
6363
using Omegac0Cands = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi>>;
6464
using Omegac0CandsKF = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfOmegacKf>>;
6565
using Omegac0CandsMc = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfToOmegaPiMCRec>>;
66-
using OmegaC0CandsMcKF = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfOmegacKf, aod::HfToOmegaPiMCRec>>;
66+
using Omegac0CandsMcKF = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfOmegacKf, aod::HfToOmegaPiMCRec>>;
6767
using Omegac0CandsMl = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfMlSelOmegacToOmegaPi>>;
6868
using Omegac0CandsMlKF = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfMlSelOmegacToOmegaPi, aod::HfOmegacKf>>;
6969
using Omegac0CandsMlMc = soa::Filtered<soa::Join<aod::HfCandToOmegaPi, aod::HfSelToOmegaPi, aod::HfMlSelOmegacToOmegaPi, aod::HfToOmegaPiMCRec>>;
@@ -518,7 +518,7 @@ struct HfTaskOmegac0ToOmegaPiQa {
518518
/// MC with KFParticle //
519519
////////////////////////////////////
520520

521-
void processMcKFParticle(OmegaC0CandsMcKF const& omegaC0CandidatesMcKF,
521+
void processMcKFParticle(Omegac0CandsMcKF const& omegaC0CandidatesMcKF,
522522
Omegac0Gen const& mcParticles)
523523
{
524524
processMc<true, false>(omegaC0CandidatesMcKF, mcParticles);
@@ -532,7 +532,7 @@ struct HfTaskOmegac0ToOmegaPiQa {
532532
}
533533
PROCESS_SWITCH(HfTaskOmegac0ToOmegaPiQa, processMcKFParticleMl, "Process MC with KFParticle, ML selections", false);
534534

535-
void processMcKFParticleFT0M(OmegaC0CandsMcKF const& omegaC0CandidatesMcKF,
535+
void processMcKFParticleFT0M(Omegac0CandsMcKF const& omegaC0CandidatesMcKF,
536536
Omegac0Gen const& mcParticles,
537537
CollisionsWithMcLabels const& collisions,
538538
McCollisionsWithFT0M const& mcCollisions)

0 commit comments

Comments
 (0)