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

---

RecallAtN.h

Go to the documentation of this file.

#ifndef Impala_Core_Training_RecallAtN_h
#define Impala_Core_Training_RecallAtN_h

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

namespace Impala
{
namespace Core
{
namespace Training
{


class RecallAtN : public Evaluation
{
public:

    RecallAtN(Table::AnnotationTable* annotation, int n, int totalRelevance) :
        Evaluation(annotation)
    {
        mN = n;
        mTotalRelevance = totalRelevance;
    }

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

        int positiveCount = 0;
        for (int i=0 ; i<table->Size() && i<mN ; i++)
        {
            Quid q = table->Get1(i);
            if (mAnnotation->IsPositive(q))
                positiveCount++;
        }
        double recall = ((double)positiveCount)/((double)mTotalRelevance);
        return recall;
    }

private:

    int mN; // N is the number at which we compute the precision.
    int mTotalRelevance;

};

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

#endif

---