Train A Model In Python