|| || || || || || || ||

---

AveragePrecision.h

Go to the documentation of this file.

#ifndef Impala_Core_Table_AveragePrecision_h
#define Impala_Core_Table_AveragePrecision_h

#include "Core/Table/AnnotationTable.h"

namespace Impala
{
namespace Core
{
namespace Table
{


template <class T>
inline Real64
AveragePrecision(T* rank, AnnotationTable* groundTruth, int topN = -1)
{
    ILOG_VAR(Impala.Core.Table.AveragePrecision);
    Real64 ap = 0;
    int positiveCount = 0;
    if (topN == -1)
        topN = rank->Size();
    for (int i=0 ; i<topN ; i++)
    {
        Quid q = rank->Get1(i);
        if (groundTruth->IsPositive(q))
        {
            positiveCount++;
            double precision = ((double)positiveCount)/((double)(i+1));
            ap += precision;
        }
    }
    if (groundTruth->GetNrPositive() > 0)
        ap /= groundTruth->GetNrPositive();
    return ap;
}

template <class T>
inline Real64
AveragePrecisionJudged(T* rank, AnnotationTable* groundTruth,
                       bool skipIsNegative, int topN = -1,
                       bool apOfFoundOnly = false)
{
    ILOG_VAR(Impala.Core.Table.AveragePrecision);
    Real64 ap = 0;
    int positiveCount = 0;
    int curRank = 0;
    if (topN == -1)
        topN = rank->Size();
    for (int i=0 ; i<topN ; i++)
    {
        Quid q = rank->Get1(i);
        int qIndex = groundTruth->GetIndex(q);
        if (qIndex == groundTruth->Size()) // not annotated
            continue;
        if (groundTruth->IsPositive(qIndex))
        {
            positiveCount++;
            double precision = ((double)positiveCount)/((double)(curRank+1));
            ap += precision;
        }
        if (groundTruth->IsPositive(qIndex) || skipIsNegative
            || groundTruth->IsNegative(qIndex))
        {
            curRank++;
        }
    }
    int factor = (apOfFoundOnly) ? positiveCount : groundTruth->GetNrPositive();
    if (factor > 0)
        ap /= factor;
    return ap;
}

} // namespace Table
} // namespace Core
} // namespace Impala

#endif

---