MIDAS

The code corresponding to the paper “Trajectory Imputation in Multi-Agent Sports with Derivative-Accumulating Self-Ensemble"

The code is written with PyTorch v1.10.1 (Python 3.8.18).

Setup

Install the requirements.txt file using the following command:
```
$ pip install -r requirements.txt
```

Training Run the following .sh file to train our proposed MIDAS model on soccer dataset:
```
$ sh scripts/midas_soccer.sh
```
Evaluating

Run the model_eval.ipynb code to evaluate trained model performance.

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
datatools		datatools
img		img
models		models
scripts		scripts
set_transformer		set_transformer
.gitignore		.gitignore
README.md		README.md
dataset.py		dataset.py
model_eval.ipynb		model_eval.ipynb
pitch_control.ipynb		pitch_control.ipynb
train.py		train.py