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

---

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::AnnotationTableBaseType* table)
    {
        Sort(table, 2, false);
        return SubCompute(table);
    }

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

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

---