by Ruochen Zhang and Carsten Eickhoff

This repository contains source code necessary to collect the dataset and reproduce the results presented in SOCCER: An Information-Sparse Discourse State Tracking Collection in the Sports Commentary Domain.

Please install the prerequisites via

pip install requirements.txt

You will also need to download the WebDriver that matches your Chrome browser version.

To collect the dataset

python src/get_dataset.py -dp=<chromedriver path>

The SOCCER dataset will be stored in thedata/ folder with train, validation and test splits.

To reproduce the team-level results of the GRU classifier

cd model/GRU_classifier

./train_eval.sh

To reproduce the team-level and player-level results of the GPT-2 variant

cd model/GPT_variant

./train_eval.sh

If you would like to cite this work, please refer to:

@inproceedings{zhang-eickhoff-2021-soccer,
    title = "{SOCCER}: An Information-Sparse Discourse State Tracking Collection in the Sports Commentary Domain",
    author = "Zhang, Ruochen  and
      Eickhoff, Carsten",
    booktitle = "Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies",
    month = jun,
    year = "2021",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2021.naacl-main.342",
    pages = "4325--4333",
}