All code for running the baseline models, as well as some helper functions for interacting with the dataset are provided in our Github repository.

For convenience, we also provide Google Colab links to examples of how to run the models: