Add CLabels::get_subset_stack()

shogun-toolbox · Mar 23, 2017 · c58ae3a · c58ae3a
1 parent 5b126ce
commit c58ae3a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/shogun/labels/Labels.cpp b/src/shogun/labels/Labels.cpp
@@ -56,6 +56,12 @@ void CLabels::remove_all_subsets()
 	m_subset_stack->remove_all_subsets();
 }
 
+CSubsetStack* CLabels::get_subset_stack()
+{
+	SG_REF(m_subset_stack);
+	return m_subset_stack;
+}
+
 float64_t CLabels::get_value(int32_t idx)
 {
 	ASSERT(m_current_values.vector && idx < get_num_labels())

diff --git a/src/shogun/labels/Labels.h b/src/shogun/labels/Labels.h
@@ -96,6 +96,11 @@ class CLabels : public CSGObject
 	 * Calls subset_changed_post() afterwards */
 	virtual void remove_all_subsets();
 
+	/**
+	 * @return subset stack
+	 */
+	virtual CSubsetStack* get_subset_stack();
+
 	/** set the confidence value for a particular label
 	 *
 	 * @param value value to set