Add MulticlassLabels constructor from BinaryLabels

shogun-toolbox · Jul 5, 2017 · fdfef65 · fdfef65
1 parent 1ee49a8
commit fdfef65
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/shogun/labels/MulticlassLabels.cpp b/src/shogun/labels/MulticlassLabels.cpp
@@ -25,6 +25,15 @@ CMulticlassLabels::CMulticlassLabels(CFile* loader) : CDenseLabels(loader)
 	init();
 }
 
+CMulticlassLabels::CMulticlassLabels(CBinaryLabels* labels)
+    : CDenseLabels(labels->get_num_labels())
+{
+	init();
+
+	for (index_t i = 0; i < labels->get_num_labels(); ++i)
+		m_labels[i] = (labels->get_label(i) == 1 ? 1 : 0);
+}
+
 CMulticlassLabels::~CMulticlassLabels()
 {
 }

diff --git a/src/shogun/labels/MulticlassLabels.h b/src/shogun/labels/MulticlassLabels.h
@@ -57,6 +57,14 @@ class CMulticlassLabels : public CDenseLabels
 		 */
 		CMulticlassLabels(CFile* loader);
 
+		/**
+		 * Convert binary labels to multiclass labels,
+		 * namely -1 is mapped to 0 and 1 to 1.
+		 *
+		 * @param labels Binary labels
+		 */
+		CMulticlassLabels(CBinaryLabels* labels);
+
 		/** destructor */
 		~CMulticlassLabels();