+ problem type macro in Machine and + return PT in QDA

shogun-toolbox · May 22, 2012 · 09e6617 · 09e6617
1 parent d7d1505
commit 09e6617
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/shogun/machine/Machine.h b/src/shogun/machine/Machine.h
@@ -99,6 +99,8 @@ enum EProblemType
 	PT_MULTICLASS = 2
 };
 
+#define MACHINE_PROBLEM_TYPE(PT) virtual EProblemType get_machine_problem_type() const { return PT; }
+
 /** @brief A generic learning machine interface.
  *
  * A machine takes as input CFeatures and (optionally) CLabels.

diff --git a/src/shogun/multiclass/QDA.h b/src/shogun/multiclass/QDA.h
@@ -36,6 +36,8 @@ namespace shogun
 class CQDA : public CNativeMulticlassMachine
 {
 	public:
+		MACHINE_PROBLEM_TYPE(PT_MULTICLASS)
+
 		/** constructor
 		 *
 		 * @param tolerance tolerance used in training