| 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 |