added a proper error message if number of labels are different

shogun-toolbox · Jul 8, 2012 · 1bc6e55 · 1bc6e55
1 parent 7cfd299
commit 1bc6e55
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/shogun/evaluation/ContingencyTableEvaluation.cpp b/src/shogun/evaluation/ContingencyTableEvaluation.cpp
@@ -84,7 +84,12 @@ void CContingencyTableEvaluation::compute_scores(CBinaryLabels* predicted, CBina
 	ASSERT(ground_truth->get_label_type() == LT_BINARY);
 	ASSERT(predicted->get_label_type() == LT_BINARY);
 
-	ASSERT(predicted->get_num_labels()==ground_truth->get_num_labels());
+	if (predicted->get_num_labels()!=ground_truth->get_num_labels())
+	{
+		SG_ERROR("%s::compute_scores(): Number of predicted labels (%d) is not "
+				"equal to number of ground truth labels (%d)!\n", get_name(),
+				predicted->get_num_labels(), ground_truth->get_num_labels());
+	}
 	m_TP = 0.0;
 	m_FP = 0.0;
 	m_TN = 0.0;