Как вычислить f1 в тензорном потоке

Question

У меня есть код, который вычисляет точность, но теперь я хочу вычислить оценку f1.

accuracy_1 = tf.reduce_mean(tf.cast(tf.equal(
    tf.argmax(output_1, axis=-1),
    tf.argmax(y_1, axis=-1)), tf.float32), name="accuracy_1")
accuracy_2 = tf.reduce_mean(tf.cast(tf.equal(
    tf.argmax(output_2, axis=-1),
    tf.argmax(y_2, axis=-1)), tf.float32), name="accuracy_2")

Как вычислить эквивалент f1 для приведенного выше кода. Я нахожу это трудным, поскольку я очень плохо знаком с tenorflow.