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

---

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::AnnotationTableBaseType* 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; // sorry not really descriptive name: N is the number at which we compute the precision.
    int mTotalRelevance;
};

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

#endif

---