I\'m following https://www.tensorflow.org/tutorials/estimator/boosted_trees_model_understanding
To train an estimator with train_in_memory=false, we use input functio