def metrics(logits, labels):
    argmax_logits = tf.argmax(logits, axis=1)
    argmax_labels = tf.argmax(labels, axis=1)
    return {'accuracy': tf.metrics.accuracy(argmax_labels, argmax_logits)}
def metrics(logits, labels):
    argmax_logits = tf.argmax(logits, axis=1)
    argmax_labels = tf.argmax(labels, axis=1)
    return {'accuracy': tf.metrics.accuracy(argmax_labels, argmax_logits)}