A more comprehensive model that covers multiple trajectory tasks and inlcudes three built sports game datasets, are released in UniTraj. Please refer to that repo for datasets and models.
Code of paper: "Uncovering the Missing Pattern: Unified Framework Towards Trajectory Imputation and Prediction" (CVPR 2023)
Yi Xu, Armin Bazarjani, Hyung-gun Chi, Chiho Choi, Yun Fu
Currently under legal review and waiting for the approval from company side, coming soon!
Please leave your institute email address in the opened issue, I can send an unofficial version of codes.