Tensorflow Estimator Predict Example