player-embedding-ice-hockey

My code is based on python 2. Check requirements.txt for a detail introduction to the python packages. I recommend python virtual environment for building this project.

Pre-Processing data

This work applies the play-by-play data from Sportlogiq. Please find a detail description of this data from my thesis (published soon). Run run_preprocess.py which include game history into training data.
To run this code, please prepare you own dataset.

Model Training

In the following training file should specify:

source_data_dir # you source data (before pre-preprocessing)
data_store_dir # your training data (after pre-preprocessing)

VaRLEA Training

The implementation of VaRLEA model is varlea.
Run train_varlea_ice_hockey.py.

CVRNN Training

The implementation of CVRNN model is cvrnn.
Run train_cvrnn_ice_hockey.py.

The model parameters are recorded in here.

Model Testing

Player Identification

run validate_player_id