register m_num_classes of MulticlassStrategy.

shogun-toolbox · Apr 28, 2012 · 1981a90 · 1981a90
1 parent 022c18d
commit 1981a90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/shogun/multiclass/MulticlassStrategy.cpp b/src/shogun/multiclass/MulticlassStrategy.cpp
@@ -17,6 +17,7 @@ using namespace shogun;
 CMulticlassStrategy::CMulticlassStrategy()
 	:m_train_labels(NULL), m_orig_labels(NULL), m_train_iter(0)
 {
+    SG_ADD(&m_num_classes, "num_classes", "Number of classes", MS_NOT_AVAILABLE);
 }
 
 void CMulticlassStrategy::train_start(CLabels *orig_labels, CLabels *train_labels)