PhiPiPi Dataset for our paper "Event Generation and Consistence Test for Physics with Sliced Wasserstein Distance"

The PhiPiPi dataset is used to train and test our fast event simulation model for our paper  "Event Generation and Consistence Test for Physics with Sliced Wasserstein Distance" in https://arxiv.org/abs/2310.17897.  The dataset is generated by BES III simulation framework to simulate phase-space Monte Carlo events, focusing on the production of psi(2S) in electron-positron annihilation. This intricate process entails the psi(2S) decay into a trio of particles: phi, pi+, and pi-, with the phi particle subsequently decaying into a K+ and K- pair.
We have meticulously partitioned the data into distinct sets for training, validation, and testing to facilitate a robust machine learning workflow. Each set is systematically saved in the convenient npy format for ease of access and integration.
The dataset contains the four-momentum components (px, py, pz, E) of the final-state particles: K+, K-, pi+, and pi-. This comprehensive inclusion ensures a detailed representation of the kinematic properties of the particles for accurate model training.
Researchers and practitioners are invited to use this dataset by incorporating it into our public code repository: https://github.com/caihao/SWD-EvtGen. Doing so will enable the training of our fast event simulation model, enhancing the fidelity and efficiency of high energy physics simulations.