metadata
license: cc-by-4.0
CausalTraj models are trained for joint multi-agent trajectory forecasting in team sports.
This repo contains 2 variants of of model weights: CausalTraj (Causal PointNet) and CausalTraj (Mamba2), both trained on the NBA SportVU dataset.
You can find the corresponding code base to load the models here: https://github.com/wezteoh/causaltraj