JetNet
Description
Gluon (g
), Top Quark (t
), Light Quark (q
), W boson (w
), and Z boson (z
) jets of ~1 TeV transverse momentum (
Each file has particle_features
; and jet_features
; arrays, containing the list of particles' features per jet and the corresponding jet's features, respectively.
particle_features
is of shape [N, 30, 4], where
- N is the total number of jets,
- 30 is the max number of particles per jet, and
- 4 is the number of particle features, in order: [
, , , mask]. See [1, Sec. 2] for definitions of these.
jet_features
is of shape [N, 4], where 4 is the number of jet features, in order: [
N = 177252, 177945, 170679, 177172, 176952 for g, t, q, w, z jets respectively.
Finally, the JetNet package is recommended for easy access and use of this dataset.
[1] Kansal et. al., Particle Cloud Generation with Message Passing Generative Adversarial Networks, NeurIPS 2021 arXiv:2106.11535