Function ndcg

benchmark/quality-eval.ts:47–53

(retrieved: string[], relevant: Set<string>, k: number)

}

function ndcg(retrieved: string[], relevant: Set<string>, k: number): number {
  const actualRelevances = retrieved.slice(0, k).map(id => relevant.has(id));
  const idealRelevances = Array.from({ length: Math.min(k, relevant.size) }, () => true);
  const idealDCG = dcg(idealRelevances, k);
  if (idealDCG === 0) return 0;
  return dcg(actualRelevances, k) / idealDCG;
}

function recall(retrieved: string[], relevant: Set<string>, k: number): number {
  if (relevant.size === 0) return 1;