add parsing of labels - Githubissues

graphnet-team / graphnet

A Deep learning library for neutrino telescopes

https://graphnet-team.github.io/graphnet/

Apache License 2.0

85 stars 86 forks source link

Closed Aske-Rosted closed 4 months ago

Aske-Rosted commented 5 months ago

This pull request allows for proper parsing of labels in the dataset class, such that the new "Direction" (https://github.com/graphnet-team/graphnet/blob/95309556cfd46a4046bc4bd7609888aab649e295/src/graphnet/training/labels.py#L29) can be instantiated correctly from a dataset config by adding the following.

labels: direction: arguments: azimuth_key: azimuth zenith_key: zenith class_name: Direction

to a .yml config file.

RasmusOrsoe commented 5 months ago

@Aske-Rosted If you update this branch with the latest from main, the tests should pass. #656

RasmusOrsoe commented 4 months ago

@Aske-Rosted please update this with latest from main before you merge