Event format#
Every dataset returns an event as a flat dict of NumPy arrays with one row per point, and every model reads a packed batch of such events.
One event#
Point-aligned fields have one row per point; scalar quantities keep a trailing axis of length 1. A PILArNet-M event from PILArNetH5Dataset contains:
Key |
Shape |
Content |
|---|---|---|
|
(N, 3) float32 |
point position |
|
(N, 1) float32 |
deposited energy |
|
(N, 1) |
semantic class |
|
(N, 1) |
particle class |
|
(N, 1) |
particle id within the event |
|
(N, 1) |
interaction id within the event |
|
(N, 1) |
background flag |
|
(N, 1) |
momentum of the point’s particle (GeV) |
|
(N, 3) |
interaction vertex of the point’s particle |
|
(N, 1) |
primary-particle flag |
|
strings |
bookkeeping |
Code |
|
|
|---|---|---|
0 |
shower |
photon |
1 |
track |
electron |
2 |
Michel |
muon |
3 |
delta |
pion |
4 |
low-energy deposit |
proton |
5 |
— |
none (low-energy deposit) |
-1 marks points to ignore. PILArNet-M coordinates lie in a cube of side 768; recipes rescale them, as described in Transforms. Other readers emit the same core keys (coord, energy, segment, instance) plus their own; see Supported detectors.
See one#
PILArNet-M-mini test event 0 after the Panda transforms: 1,175 points, colored by their true labels. Drag to rotate, scroll to zoom, and switch between semantic and particle classes.
A batch#
After the transforms, Collect keeps the keys you name and joins feat_keys into feat. collate_fn concatenates the events and builds offset:
Key |
Shape |
Content |
|---|---|---|
|
(N_total, 3) |
transformed positions |
|
(N_total, 3) int |
grid cells, when the pipeline grid-samples |
|
(N_total, C) |
features in |
|
(N_total,) |
training target, when present |
|
(B,) |
cumulative end index of each event; |
Inside a model, Point(batch) wraps the dict and derives the per-point event index from offset.