Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 1.09 KB

README.md

File metadata and controls

25 lines (18 loc) · 1.09 KB
Raw

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/