Skip to content

Zenith-angle dependent efficiencies#61

Merged
GernotMaier merged 9 commits into
mainfrom
zenith-efficiencies
May 17, 2026
Merged

Zenith-angle dependent efficiencies#61
GernotMaier merged 9 commits into
mainfrom
zenith-efficiencies

Conversation

@GernotMaier
Copy link
Copy Markdown
Member

No description provided.

@GernotMaier GernotMaier self-assigned this May 16, 2026
@GernotMaier GernotMaier requested a review from Copilot May 17, 2026 10:43
@GernotMaier GernotMaier marked this pull request as ready for review May 17, 2026 10:44
Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds zenith-angle (per-zenith-bin) efficiency outputs to the classification training pipeline and updates the plotting script to visualize those zenith-binned efficiencies (with TMVA overlays becoming optional).

Changes:

  • Extend evaluation_efficiency() to optionally compute and return per-zenith efficiency curves and persist them as efficiency_zeN in trained model configs.
  • Update the performance-metrics plotting script to select zenith bins from joblib outputs, handle missing TMVA ROOT overlays gracefully, and write per-bin plots.
  • Adjust the SHAP caching test to match the updated evaluation_efficiency() return shape.

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 6 comments.

File Description
tests/test_train_classification_shap.py Updates the mocked evaluation_efficiency() return value to a (df, dict) tuple.
src/eventdisplay_ml/scripts/plot_classification_performance_metrics.py Adds zenith-bin discovery/selection and optional TMVA overlays; generates per-zenith output plots.
src/eventdisplay_ml/models.py Persists overall and per-zenith efficiency DataFrames into the model config (efficiency, efficiency_zeN).
src/eventdisplay_ml/evaluate.py Refactors efficiency computation into a helper and adds optional per-zenith efficiency calculation/return.

Comment thread src/eventdisplay_ml/scripts/plot_classification_performance_metrics.py Outdated
Comment thread src/eventdisplay_ml/scripts/plot_classification_performance_metrics.py Outdated
Comment thread src/eventdisplay_ml/scripts/plot_classification_performance_metrics.py Outdated
Comment thread src/eventdisplay_ml/scripts/plot_classification_performance_metrics.py Outdated
Comment thread src/eventdisplay_ml/evaluate.py
Comment thread src/eventdisplay_ml/evaluate.py
GernotMaier and others added 4 commits May 17, 2026 12:47
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
@GernotMaier GernotMaier merged commit 9a9ddd1 into main May 17, 2026
2 checks passed
@GernotMaier GernotMaier deleted the zenith-efficiencies branch May 17, 2026 10:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants