Updated one more include

shogun-toolbox · Jul 18, 2012 · 46b4d2c · 46b4d2c
1 parent 5712c9a
commit 46b4d2c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/examples/undocumented/libshogun/transfer_multitasklogisticregression.cpp b/examples/undocumented/libshogun/transfer_multitasklogisticregression.cpp
@@ -2,6 +2,7 @@
 #include <shogun/features/DenseFeatures.h>
 #include <shogun/lib/IndexBlock.h>
 #include <shogun/lib/IndexBlockGroup.h>
+#include <shogun/lib/IndexBlockTree.h>
 #include <shogun/transfer/multitask/MultitaskLogisticRegression.h>
 #include <shogun/base/init.h>
 #include <shogun/lib/common.h>
@@ -34,7 +35,6 @@ int main(int argc, char** argv)
 
 	CIndexBlock* first_task = new CIndexBlock(0,2);
 	CIndexBlock* second_task = new CIndexBlock(2,4);
-
 	CIndexBlockGroup* task_group = new CIndexBlockGroup();
 	task_group->add_block(first_task);
 	task_group->add_block(second_task);
@@ -44,6 +44,18 @@ int main(int argc, char** argv)
 
 	regressor->set_current_task(0);
 	regressor->get_w().display_vector();
+
+	CIndexBlock* root_task = new CIndexBlock(0,4);
+	root_task->add_sub_block(first_task);
+	root_task->add_sub_block(second_task);
+	CIndexBlockTree* task_tree = new CIndexBlockTree(root_task);
+
+	regressor->set_task_relation(task_tree);
+	regressor->train();
+
+	regressor->set_current_task(0);
+	regressor->get_w().display_vector();
+
 	SG_UNREF(regressor);
 	exit_shogun();
 	return 0;

diff --git a/src/shogun/lib/slep/slep_solver.cpp b/src/shogun/lib/slep/slep_solver.cpp
@@ -8,7 +8,7 @@
  * Copyright (C) 2010-2012 Jun Liu, Jieping Ye
  */
 
-#include <shogun/lib/slep/slep_logistic.h>
+#include <shogun/lib/slep/slep_solver.h>
 #include <shogun/mathematics/Math.h>
 #include <shogun/lib/slep/q1/eppMatrix.h>
 #include <shogun/lib/slep/q1/eppVector.h>