Commit d745e4f0 authored by Katherine Wu's avatar Katherine Wu Committed by TensorFlower Gardener
Browse files

Add recall and precision at K to metrics.

This change adds the options to set top_k and class_id to the Recall and Precision metric classes. The top_k is computed for dense labels, which differs from tf.metrics.recall_at_k, which expects sparse labels.

PiperOrigin-RevId: 228944786
parent 03f9fc0b
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment