Skip to content

Draft: Add heterogeneous input support

Dmitrii Kobylianskii requested to merge dkobylia-gn2-hetero into main

This is draft MR to review/discuss changes before the final MR to main repo

Idea:
Create an abstract class for a constituent loader with the getData function which returns a vector of values. Each loader (e.g., TracksLoader, IParticlesLoader, ...) is inherited from it. Then we have a vector of pointers and sequentially apply all of them and feed output to GNN.

TO-DOs:

  • Add abstract class for constituent loading
  • Add tracksLoader class -- Fully works as a replacement
  • Add tracksLoader configuration
  • Add IParticleLoader class -- WIP
  • Add electronLoader class
  • Add subjetLoader class
  • Add flexible configuration of different constituents
Edited by Dmitrii Kobylianskii

Merge request reports