python code for ensemble machine learning model