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

---

AveragePrecision.h

Go to the documentation of this file.

#ifndef Impala_Core_Training_AveragePrecision_h
#define Impala_Core_Training_AveragePrecision_h

#include "Core/Training/Evaluation.h"
#include "Core/Table/Sort.h"
#include "Core/Table/Reverse.h"

namespace Impala
{
namespace Core
{
namespace Training
{

class AveragePrecision : public Evaluation
{
public:

    AveragePrecision(Table::AnnotationTable* annotation) :
        Evaluation(annotation)
    {
    }

    virtual double
    Compute(Table::ScoreTable* table)
    {
        Sort(table, 2, false);
        return ComputeSub(table);
    }

    virtual double
    ComputeReversed(Table::ScoreTable* table)
    {
        Sort(table, 2, false);
        Reverse(table);
        return ComputeSub(table);
    }

private:

    double
    ComputeSub(Table::ScoreTable* table)
    {
        double ap = 0;
        int positiveCount = 0;
        for (int i=0 ; i<table->Size() ; i++)
        {
            Quid q = table->Get1(i);
            if (mAnnotation->IsPositive(q))
            {
                positiveCount++;
                double precision = ((double)positiveCount)/((double)(i+1));
                ap += precision;
            }
        }
        if (positiveCount > 0)
            ap /= (double)positiveCount;
        return ap;
    }

    ILOG_VAR_DECL;

};

ILOG_VAR_INIT(AveragePrecision, Impala.Core.Training);

}//namespace Core
}//namespace Training
}//namespace Impala

#endif

---