NICE_VisLearning/cbaselib/ClassificationResults.h

/** 
* @file ClassificationResults.h
// refactor-nice.pl: check this substitution
// old: * @brief vector of ClassificationResult's
* @brief std::vector of ClassificationResult's
* @author Erik Rodner
* @date 02/13/2008

*/
#ifndef CLASSIFICATIONRESULTSINCLUDE
#define CLASSIFICATIONRESULTSINCLUDE

#include <vislearning/nice_nonvis.h>

#include <vector>
#include "ClassificationResult.h"

namespace OBJREC {

/** vector of ClassificationResult's
    @see ClassificationResult */
class ClassificationResults : public std::vector<ClassificationResult>
{
	public:
		enum {
			PERF_AVG_PRECISION_11_POINT = 0,
			PERF_AUC = 1,
			PERF_AVG_PRECISION
		};

    protected:
		

    public:
  
	/** simple constructor */
	ClassificationResults();
      
	/** simple destructor */
	virtual ~ClassificationResults();
   
	/** write BINARY classification results in WEKA format
	    suitable for drawing ROC curves 
	    @param filename write results to filename
	    @param classno class number of object class within the binary classification setting
	*/
	void writeWEKA ( const std::string & filename, int classno ) const;

	/** please choose the type from the public enum,
	 *  this only works for binary classification problems */
	double getBinaryClassPerformance ( int type = PERF_AUC ) const;
};


} // namespace

#endif