tf.estimator.PoissonRegressionHead