Module: tfr.keras.metrics  |  TensorFlow Ranking (original) (raw)

Keras metrics in TF-Ranking.

Classes

class ARPMetric: Average relevance position (ARP).

class AlphaDCGMetric: Alpha discounted cumulative gain (alphaDCG).

class DCGMetric: Discounted cumulative gain (DCG).

class HitsMetric: Hits@k metric.

class MRRMetric: Mean reciprocal rank (MRR).

class MeanAveragePrecisionMetric: Mean average precision (MAP).

class NDCGMetric: Normalized discounted cumulative gain (NDCG).

class OPAMetric: Ordered pair accuracy (OPA).

class PrecisionIAMetric: Precision-IA@k (Pre-IA@k).

class PrecisionMetric: Precision@k (P@k).

class RankingMetricKey: Ranking metric key strings.

class RecallMetric: Recall@k (R@k).

Functions

default_keras_metrics(...): Returns a list of ranking metrics.

get(...): Factory method to get a list of ranking metrics.