Added reject number counting

shogun-toolbox · Mar 10, 2012 · f9907d2 · f9907d2
1 parent df59536
commit f9907d2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/shogun/evaluation/MulticlassAccuracy.cpp b/src/shogun/evaluation/MulticlassAccuracy.cpp
@@ -40,6 +40,7 @@ float64_t CMulticlassAccuracy::evaluate(CLabels* predicted, CLabels* ground_trut
 			else if (predicted_label==ground_truth->get_int_label(i))
 				correct++;
 		}
+		m_rejects_num = length-total;
 		SG_DEBUG("correct=%d, total=%d, rejected=%d\n",correct,total,length-total);
 		return ((float64_t)correct)/total;
 	}

diff --git a/src/shogun/evaluation/MulticlassAccuracy.h b/src/shogun/evaluation/MulticlassAccuracy.h
@@ -36,10 +36,10 @@ class CMulticlassAccuracy: public CEvaluation
 public:
 	/** constructor */
 	CMulticlassAccuracy() : 
-		CEvaluation(), m_ignore_rejects(false) {};
+		CEvaluation(), m_ignore_rejects(false), m_rejects_num(0) {};
 
 	CMulticlassAccuracy(bool ignore_rejects) : 
-		CEvaluation(), m_ignore_rejects(ignore_rejects) {};
+		CEvaluation(), m_ignore_rejects(ignore_rejects), m_rejects_num(0) {};
 
 	/** destructor */
 	virtual ~CMulticlassAccuracy() {};
@@ -63,13 +63,22 @@ class CMulticlassAccuracy: public CEvaluation
 		return ED_MAXIMIZE;
 	}
 
+	/** get rejects num */
+	int32_t get_rejects_num() const 
+	{
+		return m_rejects_num;
+	}
+
 	/** get name */
 	virtual inline const char* get_name() const { return "MulticlassAccuracy"; }
 
 protected:
 
 	/** ignore rejects */
 	bool m_ignore_rejects;
+
+	/** rejects num */
+	int32_t m_rejects_num;
 };
 
 }