tf.estimator.BinaryClassHead