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

---

MakeEvaluation.h

Go to the documentation of this file.

#ifndef Impala_Core_Training_MakeEvaluation_h
#define Impala_Core_Training_MakeEvaluation_h

#include "Basis/String.h"
#include "Core/Training/AveragePrecision.h"
#include "Core/Training/BalancedAveragePrecision.h"
#include "Core/Training/AreaUnderRocCurve.h"
#include "Core/Training/PrecisionAtN.h"
#include "Core/Training/RecallAtN.h"

namespace Impala
{
namespace Core
{
namespace Training
{

    Evaluation* MakeEvaluation(const std::string& description,
                               Table::AnnotationTable* annotation)
    {
        ILOG_VAR(Impala.Core.Training.MakeEvaluation);

        if(description == "ap" || 
           description == "AP" ||
           description == "AveragePrecision")
           return new AveragePrecision(annotation);
        if(description == "bap" || 
           description == "BAP" ||
           description == "BalancedAveragePrecision")
           return new BalancedAveragePrecision(annotation);
        if(description == "auc" || 
           description == "AUC" ||
           description == "AreaUnderRocCurve")
           return new AreaUnderRocCurve(annotation);
        if(description.substr(0,2) == "P@")
        {
            int n = atoi(description.substr(2));
            return new PrecisionAtN(annotation, n);
        }
        if(description.substr(0,10) == "precision@")
        {
            int n = atoi(description.substr(10));
            return new PrecisionAtN(annotation, n);
        }
        if(description.substr(0,2) == "R@")
        {
            int n = atoi(description.substr(2));
            int totalRelevance = annotation->GetNrPositive();
            return new RecallAtN(annotation, n, totalRelevance);
        }
        if(description.substr(0,10) == "recall@")
        {
            int n = atoi(description.substr(10));
            int totalRelevance = annotation->GetNrPositive();
            return new RecallAtN(annotation, n, totalRelevance);
        }
        ILOG_ERROR("description not recognised" << description);
        return new Evaluation(annotation);
    }


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

#endif

---