Build your first machine learning model in Python