Supervised Machine Learning Code Python