JetNet

Gluon (<code>g</code>), Top Quark (<code>t</code>), Light Quark (<code>q</code>), W boson (<code>w</code>), and Z boson (<code>z</code>) jets of ~1 TeV transverse momentum (\(p_T\)), as introduced in Ref. [1]. Each file has <code>particle_features</code>; and <code>jet_features</code>; arrays, containing the list of particles' features per jet and the corresponding jet's features, respectively. <code>particle_features</code> 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: [\(\eta_{\mathrm{rel}}\), \(\varphi_{\mathrm{rel}}\), \(p_{T\mathrm{rel}}\), mask]. See [1, Sec. 2] for definitions of these. <code>jet_features</code> 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., <em>Particle Cloud Generation with Message Passing Generative Adversarial Networks</em>, NeurIPS 2021 <code>arXiv:2106.11535</code>