neurovlm.metrics.recall_at_k

Contents

neurovlm.metrics.recall_at_k#

neurovlm.metrics.recall_at_k(cos_sim, k)[source]#
Parameters:
  • cos_sim (2d tensor) – Cosine similarity matrix.

  • k (int) – Top k most similar items to consider.

Returns:

recall – Recall @ k.

Return type:

float