/usr/include/shogun/evaluation/PRCEvaluation.h is in libshogun-dev 3.2.0-7.3build4.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | /*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* Written (W) 2011 Sergey Lisitsyn
* Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
*/
#ifndef PRCEVALUATION_H_
#define PRCEVALUATION_H_
#include <shogun/evaluation/BinaryClassEvaluation.h>
namespace shogun
{
class CLabels;
/** @brief Class PRCEvaluation used to evaluate PRC
* (Precision Recall Curve) and an area under PRC curve (auPRC).
*
*/
class CPRCEvaluation: public CBinaryClassEvaluation
{
public:
/** constructor */
CPRCEvaluation() :
CBinaryClassEvaluation(), m_computed(false)
{
m_PRC_graph = SGMatrix<float64_t>();
m_thresholds = SGVector<float64_t>();
m_auPRC = 0.0;
};
/** destructor */
virtual ~CPRCEvaluation();
/** get name */
virtual const char* get_name() const { return "PRCEvaluation"; };
/** evaluate PRC and auPRC
* @param predicted labels
* @param ground_truth labels assumed to be correct
* @return auPRC
*/
virtual float64_t evaluate(CLabels* predicted, CLabels* ground_truth);
inline EEvaluationDirection get_evaluation_direction() const
{
return ED_MAXIMIZE;
}
/** get auPRC
* @return area under PRC (auPRC)
*/
float64_t get_auPRC();
/** get PRC
* @return PRC graph matrix
*/
SGMatrix<float64_t> get_PRC();
/** get thresholds corresponding to points on the PRC graph
* @return thresholds
*/
SGVector<float64_t> get_thresholds();
protected:
/** 2-d array used to store PRC graph */
SGMatrix<float64_t> m_PRC_graph;
/** vector with thresholds corresponding to points on the PRC graph */
SGVector<float64_t> m_thresholds;
/** area under PRC graph */
float64_t m_auPRC;
/** indicator of PRC and auPRC being computed already */
bool m_computed;
};
}
#endif /* PRCEVALUATION_H_ */
|