Skip to content

[PWGDQ] add computation of chi2-like values from ML score #16716

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mcoquet642 merged 1 commit into from
Jun 17, 2026
+27 −6
Merged

[PWGDQ] add computation of chi2-like values from ML score #16716

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 27 additions & 6 deletions PWGDQ/Tasks/qaMatching.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1999,6 +1999,21 @@ struct QaMatching {
return attempts;
}

template <class TMUON>
void getMatchChi2AndScore(TMUON const& muonTrack, float& matchChi2, float& matchScore)
{
matchChi2 = muonTrack.chi2MatchMCHMFT() / MatchingDegreesOfFreedom;
matchScore = muonTrack.matchScoreMCHMFT();
if (matchScore >= 0 && matchChi2 < 0) {
// match score from ML-based matching, we compute a chi2-like value from the score
float matchScoreInv = (matchScore > 0) ? 1.0 / matchScore : std::numeric_limits<float>::max();
matchChi2 = matchScoreInv - 1.f;
} else {
// we assume a standard chi2-based matching, and compute the score value from the chi2
matchScore = chi2ToScore(muonTrack.chi2MatchMCHMFT(), MatchingDegreesOfFreedom, MatchingScoreChi2Max);
}
}

template <bool isMC, class EVT, class BC, class TMUON, class TMFT>
void fillCollisions(EVT const& collisions,
BC const& bcs,
Expand Down Expand Up @@ -2052,8 +2067,10 @@ struct QaMatching {
} else {
// global muon tracks (MFT-MCH or MFT-MCH-MID)
int64_t muonTrackIndex = muonTrack.globalIndex();
double matchChi2 = muonTrack.chi2MatchMCHMFT() / MatchingDegreesOfFreedom;
double matchScore = chi2ToScore(muonTrack.chi2MatchMCHMFT(), MatchingDegreesOfFreedom, MatchingScoreChi2Max);
float matchChi2{-1};
float matchScore{-1};
getMatchChi2AndScore(muonTrack, matchChi2, matchScore);

auto const& mchTrack = muonTrack.template matchMCHTrack_as<TMUON>();
int64_t mchTrackIndex = mchTrack.globalIndex();
auto const& mftTrack = muonTrack.template matchMFTTrack_as<TMFT>();
Expand Down Expand Up @@ -2902,8 +2919,12 @@ struct QaMatching {
std::vector<float> inputML = mlResponse.getInputFeatures(muonTrack, mftTrack, mchTrack, mftTrackProp, mchTrackProp, collision);
mlResponse.isSelectedMl(inputML, 0, output);
float matchScore = output[0];
float matchChi2Prod = muonTrack.chi2MatchMCHMFT() / MatchingDegreesOfFreedom;
float matchScoreProd = chi2ToScore(muonTrack.chi2MatchMCHMFT(), MatchingDegreesOfFreedom, MatchingScoreChi2Max);
float matchScoreInv = (matchScore > 0) ? 1.0 / matchScore : std::numeric_limits<float>::max();
float matchChi2 = matchScoreInv - 1.f;

float matchChi2Prod{-1};
float matchScoreProd{-1};
getMatchChi2AndScore(muonTrack, matchChi2Prod, matchScoreProd);

// check if a vector of global muon candidates is already available for the current MCH index
// if not, initialize a new one and add the current global muon track
Expand All @@ -2917,7 +2938,7 @@ struct QaMatching {
mftTrackProp,
mchTrackProp,
matchScore,
-1,
matchChi2,
-1,
matchScoreProd,
matchChi2Prod,
Expand All @@ -2932,7 +2953,7 @@ struct QaMatching {
mftTrackProp,
mchTrackProp,
matchScore,
-1,
matchChi2,
-1,
matchScoreProd,
matchChi2Prod,
Expand Down
Loading