Generalized Random Forests Python Example