Attend the Women in ML Symposium on December 7 Register now


Stay organized with collections Save and categorize content based on your preferences.

Factory method to get a list of ranking metrics.

Example Usage:

  metric = tfr.keras.metics.get(tfr.keras.metrics.RankingMetricKey.MRR)

to get Mean Reciprocal Rank.

  metric = tfr.keras.metics.get(tfr.keras.metrics.RankingMetricKey.MRR,

to get MRR@2.

key An attribute of RankingMetricKey, defining which metric objects to return.
name Name of metrics.
dtype Dtype of the metrics.
topn Cutoff of how many items are considered in the metric.
**kwargs Keyword arguments for the metric object.

A tf.keras.metrics.Metric. See _RankingMetric signature for more details.

ValueError If key is unsupported.