JetNet

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]. 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: [(\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