Downloads · 30 days
0
wezteoh/causaltraj
causaltraj is a machine learning model from wezteoh. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
CausalTraj models are trained for joint multi-agent trajectory forecasting in team sports.
Downloads · 30 days
0
Access
Public
Updated Nov 24, 2025
Repo size
75.7 MB
Likes
0
Public
Click a slice to open those files.
.ckpt75.7 MB · 100%
From the Hugging Face model README
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