Rank-Based Evaluation

Implementation of ranked based evaluator.

class pykeen.evaluation.RankBasedMetricResults(mean_rank, mean_reciprocal_rank, hits_at_k, adjusted_mean_rank)[source]

Results from computing metrics.

Includes results from:

  • Mean Rank (MR)

  • Mean Reciprocal Rank (MRR)

  • Adjusted Mean Rank (AMR; [berrendorf2020])

  • Hits @ K

class pykeen.evaluation.RankBasedEvaluator(ks=None, filtered=True)[source]

A rank-based evaluator for KGE models.

Calculates:

  • Mean Rank (MR)

  • Mean Reciprocal Rank (MRR)

  • Adjusted Mean Rank (AMR; [berrendorf2020])

  • Hits @ K