The jupyter notebook prediction_baseline.ipynb is used in the Kaggle competition Lyft Motion Prediction for Autonomous Vehicles A modified version of the notebook is also published on Kaggle.