JetNet150

Gluon (g), Top Quark (t), Light Quark (q), W boson (w), and Z boson (z) jets of ~1 TeV transverse momentum (\(p_T\)), as introduced in Ref. [1], but up to 150 particles. 

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, 150, 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: [\(\eta_{\mathrm{rel}}\), \(\varphi_{\mathrm{rel}}\), \(p_{T\mathrm{rel}}\), 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: [\(p_T\), \(\eta\), mass, # of particles]. 

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