DSARF on Figure 8, Agentformer on Basketball

Baselines on Figure 8 and Basketball data for paper titled 'Discovering group dynamics in synchronous time series via hierarchical recurrent switching-state models'

Environment

• Tested OS: MacOS, Linux
• Python >= 3.7
• PyTorch == 1.8.0

Dependencies:

1. Install PyTorch 1.8.0 with the correct CUDA version.
2. Install the dependencies:

   pip install -r requirements.txt
   

Cloning the anonymous repo

Please follow the instructions here to clone an anonymous repo. (Credit : Clone Anonymous Github created by fedebotu)

DSARF on Figure 8

For reproducing the experiments for complete independence, multi channel and complete pooling, see the notebook provided in dsarf_on_figure_8/DSARF_on_figure_8.ipynb

AgentFormer on Basketball

For reproducing the experiments for predicting basketball player trajectories, follow the readme in agentformer_on_bball/