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

---

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::AnnotationTableBaseType* table)
    {
        ILOG_DEBUG("sorting table with size = " << table->Size());
        Sort(table, 2, false);
        ILOG_DEBUG("calling ComputeSub");
        return ComputeSub(table);
    }

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

private:
    double ComputeSub(Table::AnnotationTableBaseType* 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

---