Skip to content

Minor Changes to run Example Training Files on the NuBench dataset#907

Merged
christianlocatelli merged 1 commit into
graphnet-team:mainfrom
christianlocatelli:2026_06_29_nubench_plots
Jul 1, 2026
Merged

Minor Changes to run Example Training Files on the NuBench dataset#907
christianlocatelli merged 1 commit into
graphnet-team:mainfrom
christianlocatelli:2026_06_29_nubench_plots

Conversation

@christianlocatelli

@christianlocatelli christianlocatelli commented Jun 29, 2026

Copy link
Copy Markdown
Collaborator

I ran one of the example training files on the low-energy Hexagon Ice Dataset and ran into 3 issues that I fixed.

  1. I added the correct path to the module KNNGraph into 01_train_dynedge.py as the former path was labelled as outdated.
  2. I added support for file paths with spaces into curated_datamodule.py for people like me with messy path structures and names.
  3. I changed the order of the parameters "charge" and "t" in nubench_datasets.py, as this order is expected from the Hexagon detector's feature_map() in the detector model nubench.py.

@sevmag sevmag left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Looks Good! 🚀

@christianlocatelli christianlocatelli merged commit 92c5d6a into graphnet-team:main Jul 1, 2026
12 checks passed
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