Rejection strategies fixes

shogun-toolbox · Mar 10, 2012 · 835eb0b · 835eb0b
1 parent 0e176ef
commit 835eb0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/shogun/features/RejectionStrategy.h b/src/shogun/features/RejectionStrategy.h
@@ -52,7 +52,7 @@ class CThresholdReject : public CRejectionStrategy
 		/** get name */
 		virtual const char* get_name() const 
 		{
-				return "AllNegativesMulticlassReject";
+				return "ThresholdReject";
 		}
 
 		/** returns true if given output set leads to rejection */
@@ -68,6 +68,7 @@ class CThresholdReject : public CRejectionStrategy
 
 protected:
 
+		/** threshold */
 		float64_t m_threshold;
 
 

diff --git a/src/shogun/machine/MulticlassMachine.h b/src/shogun/machine/MulticlassMachine.h
@@ -118,6 +118,7 @@ class CMulticlassMachine : public CMachine
 		/** set rejection strategy */
 		inline void set_rejection_strategy(CRejectionStrategy* rejection_strategy)
 		{
+			SG_UNREF(m_rejection_strategy);
 			m_rejection_strategy = rejection_strategy;
 		}