Added problem type to clustering

shogun-toolbox · May 22, 2012 · 2741f89 · 2741f89
1 parent 33a03c0
commit 2741f89
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/shogun/clustering/Hierarchical.h b/src/shogun/clustering/Hierarchical.h
@@ -48,6 +48,8 @@ class CHierarchical : public CDistanceMachine
 		CHierarchical(int32_t merges, CDistance* d);
 		virtual ~CHierarchical();
 
+		MACHINE_PROBLEM_TYPE(PT_MULTICLASS);
+
 		/** get classifier type
 		 *
 		 * @return classifier type HIERARCHICAL

diff --git a/src/shogun/clustering/KMeans.h b/src/shogun/clustering/KMeans.h
@@ -50,6 +50,9 @@ class CKMeans : public CDistanceMachine
 		CKMeans(int32_t k, CDistance* d);
 		virtual ~CKMeans();
 
+
+		MACHINE_PROBLEM_TYPE(PT_MULTICLASS)
+
 		/** get classifier type
 		 *
 		 * @return classifier type KMEANS