graphnet-team / graphnet

A Deep learning library for neutrino telescopes
https://graphnet-team.github.io/graphnet/
Apache License 2.0
85 stars 85 forks source link

Add `PrometheusReader` and extractors #688

Closed RasmusOrsoe closed 3 months ago

RasmusOrsoe commented 3 months ago

This PR adds data conversion code for files generated by the open-source simulation tool Prometheus, allowing users of graphnet to convert open-source simulation to a supported backend.

The changes include:

  1. Adding a PrometheusReader
  2. Adding extractors PrometheusExtractor, PrometheusTruthExtractor and PrometheusFeatureExtractor
  3. Adding a test file under data/examples/prometheus
  4. Adding example under graphnet/examples/06_prometheus on data conversion
  5. Adding unit test of example.