Fixes for eigen3 based multitask logistic regression classes

shogun-toolbox · Jul 17, 2012 · 0771067 · 0771067
1 parent 154176c
commit 0771067
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/shogun/lib/slep/malsar_joint_feature_learning.cpp b/src/shogun/lib/slep/malsar_joint_feature_learning.cpp
@@ -203,7 +203,7 @@ slep_result_t malsar_joint_feature_learning(
 	SGMatrix<float64_t> tasks_w(n_feats, n_tasks);
 	for (int i=0; i<n_feats; i++)
 	{
-		for (int task=0; task<n_tasks; task++)
+		for (task=0; task<n_tasks; task++)
 			tasks_w[i] = Wzp(i,task);
 	}
 	SGVector<float64_t> tasks_c(n_tasks);

diff --git a/src/shogun/transfer/multitask/MultitaskL1L2LogisticRegression.cpp b/src/shogun/transfer/multitask/MultitaskL1L2LogisticRegression.cpp
@@ -72,7 +72,7 @@ bool CMultitaskL1L2LogisticRegression::train_machine(CFeatures* data)
 	m_tasks_c = model.c;
 #else
 	SG_WARNING("Please install Eigen3 to use MultitaskL1L2LogisticRegression\n");
-	m_tasks_w = SGMatrix<float64_t>(features->get_dim_features_space(), options.n_tasks); 
+	m_tasks_w = SGMatrix<float64_t>(((CDotFeatures*)features)->get_dim_feature_space(), options.n_tasks); 
 	m_tasks_c = SGVector<float64_t>(options.n_tasks); 
 #endif
 

diff --git a/src/shogun/transfer/multitask/MultitaskTraceLogisticRegression.cpp b/src/shogun/transfer/multitask/MultitaskTraceLogisticRegression.cpp
@@ -66,7 +66,7 @@ bool CMultitaskTraceLogisticRegression::train_machine(CFeatures* data)
 	m_tasks_c = model.c;
 #else
 	SG_WARNING("Please install Eigen3 to use MultitaskTraceLogisticRegression\n");
-	m_tasks_w = SGMatrix<float64_t>(features->get_dim_features_space(), options.n_tasks); 
+	m_tasks_w = SGMatrix<float64_t>((((CDotFeatures*)features)->get_dim_feature_space(), options.n_tasks); 
 	m_tasks_c = SGVector<float64_t>(options.n_tasks); 
 #endif