I have a machine learning program that I am trying to write. Just because training the machine learning program is so time-consuming, I have decided to write it in an ipynb file