https://medium.com/@monocosmo77/updates-on-imitation-learning-2023-part5-machine-learning-db782213527c