Fixed sizes of ROC and PRC graphs

shogun-toolbox · May 8, 2012 · a2c0e99 · a2c0e99
1 parent 276fa54
commit a2c0e99
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/shogun/evaluation/PRCEvaluation.cpp b/src/shogun/evaluation/PRCEvaluation.cpp
@@ -45,7 +45,7 @@ float64_t CPRCEvaluation::evaluate(CLabels* predicted, CLabels* ground_truth)
 
 	// clean and initialize graph and auPRC
 	SG_FREE(labels);
-	m_PRC_graph = SGMatrix<float64_t>(length,2);
+	m_PRC_graph = SGMatrix<float64_t>(2,length);
 	m_thresholds = SGVector<float64_t>(length);
 	m_auPRC = 0.0;
 

diff --git a/src/shogun/evaluation/ROCEvaluation.cpp b/src/shogun/evaluation/ROCEvaluation.cpp
@@ -60,7 +60,7 @@ float64_t CROCEvaluation::evaluate(CLabels* predicted, CLabels* ground_truth)
 	SG_FREE(labels);
 
 	// initialize graph and auROC
-	m_ROC_graph = SGMatrix<float64_t>(diff_count+1,2);
+	m_ROC_graph = SGMatrix<float64_t>(2,diff_count+1);
 	m_thresholds = SGVector<float64_t>(length);
 	m_auROC = 0.0;