Function dcg

benchmark/quality-eval.ts:39–45

(relevances: boolean[], k: number)

}

function dcg(relevances: boolean[], k: number): number {
  let sum = 0;
  for (let i = 0; i < Math.min(k, relevances.length); i++) {
    sum += (relevances[i] ? 1 : 0) / Math.log2(i + 2);
  }
  return sum;
}

function ndcg(retrieved: string[], relevant: Set<string>, k: number): number {
  const actualRelevances = retrieved.slice(0, k).map(id => relevant.has(id));