https://www.technopython.com/2021/09/machine-learning-full-explanation.html