https://medium.com/@monocosmo77/updates-on-imitation-learning-2023-part3-machine-learning-bf34f1be0eea