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

---

AreaUnderRocCurve.h

Go to the documentation of this file.

#ifndef Impala_CoreTraining_AreaUnderRocCurve_h
#define Impala_CoreTraining_AreaUnderRocCurve_h

#include "Core/Training/Evaluation.h"

namespace Impala
{
namespace Core
{
namespace Training
{

class AreaUnderRocCurve : public Evaluation
{
public:

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

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

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

private:

    double
    SubCompute(Table::ScoreTable* table)
    {
        int pos = 0;
        int neg = 0;
        int area = 0;
        for (int i=0 ; i<table->Size() ; i++)
        {
            Quid q = table->Get1(i);
            if (mAnnotation->IsPositive(q))
            {
                ++pos;
            }
            else //if(mAnnotation->IsNegative(q))
            {
                area += pos;
                ++neg;
            }
        }
        return (double)area / (double)(pos*neg);
    }
};

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

#endif

---