π pip install unravelsports
The unravelsports package aims to aid researchers, analysts and enthusiasts by providing intermediary steps in the complex process of converting raw sports data into meaningful information and actionable insights.
β½ Soccer positional tracking data into Graphs to train graph neural networks by leveraging the powerful Kloppy data conversion standard for
- Metrica
- Sportec
- Tracab (CyronHego)
- SecondSpectrum
- SkillCorner
- StatsPerform
π BigDataBowl American football positional tracking data into Graphs to train graph neural networks by leveraging Polars.
These Graphs can be used with Spektral - a flexible framework for training graph neural networks.
unravelsports
allows you to randomize and split data into train, test and validation sets along matches, sequences or possessions to avoid leakage and improve model quality. And finally, train, validate and test your (custom) Graph model(s) and easily predict on new data.
β More to come soon...!
π β½ The Quick Start Jupyter Notebook explains how to convert any positional tracking data from Kloppy to Spektral GNN in a few easy steps while walking you through the most important features and documentation.
π β½ The Graph Converter Tutorial Jupyter Notebook gives an in-depth walkthrough.
π π The BigDataBowl Converter Tutorial Jupyter Notebook gives an guide on how to convert the BigDataBowl data into Graphs.
For now, follow the Graph Converter Tutorial and check the Graph FAQ, more documentation will follow!
Additional reading:
π A Graph Neural Network Deep-dive into Successful Counterattacks {A. Sahasrabudhe & J. Bekkers, 2023}
The easiest way to get started is:
pip install unravelsports
Due to compatibility issues unravelsports currently only works on Python 3.11 with:
spektral==1.20.0
tensorflow==2.14.0
keras==2.14.0
kloppy==3.15.0
polars==1.2.1
These dependencies come pre-installed with the package. It is advised to create a virtual environment.
This package is tested on the latest versions of Ubuntu, MacOS and Windows.
All contributions, bug reports, bug fixes, documentation improvements, enhancements, and ideas are welcome.
An overview on how to contribute can be found in the contributing guide.
If you use this repository for any educational purposes, research, project etc., please reference both:
π The unravelsports
package.
BibTex
@software{unravelsports2024repository, author = {Bekkers, Joris}, title = {unravelsports}, version = {0.1.0}, year = {2024}, publisher = {GitHub}, url = {https://github.com/unravelsports/unravelsports} }
BibTex
@inproceedings{sahasrabudhe2023graph, title={A Graph Neural Network deep-dive into successful counterattacks}, author={Sahasrabudhe, Amod and Bekkers, Joris}, booktitle={17th Annual MIT Sloan Sports Analytics Conference. Boston, MA, USA: MIT}, pages={15}, year={2023} }
and let me know on
This project is licensed under the Mozilla Public License Version 2.0 (MPL), which requires that you include a copy of the license and provide attribution to the original authors. Any modifications you make to the MPL-licensed files must be documented, and the source code for those modifications must be made open-source under the same license.