def two_class_accuracy(predictions, labels, weights_fn=None):
  """Accuracy for two class classification with 0/1 labels."""
  with tf.variable_scope("two_class_accuracy", values=[predictions, labels]):
    del weights_fn
    hard_predictions = tf.to_int32(tf.math.round(tf.squeeze(predictions)))
    int_labels = tf.to_int32(labels)
    _, accuracy = tf.metrics.accuracy(labels=int_labels,
                                      predictions=hard_predictions)
    return accuracy, tf.constant(1.0)
Accuracy for two class classification with 0/1 labels.
tensor2tensor/utils/metrics.py