Fix ROC error messages (shogun-toolbox#4042)

Fixes shogun-toolbox#4028
ktiefe · Jul 26, 2019 · 8d8b2a7 · 8d8b2a7
1 parent ee1adec
commit 8d8b2a7
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/shogun/evaluation/ROCEvaluation.cpp b/src/shogun/evaluation/ROCEvaluation.cpp
@@ -19,8 +19,16 @@ CROCEvaluation::~CROCEvaluation()
 
 float64_t CROCEvaluation::evaluate(CLabels* predicted, CLabels* ground_truth)
 {
-	REQUIRE(predicted->get_label_type()==LT_BINARY, "ROC evalution requires binary labels.");
-	REQUIRE(ground_truth->get_label_type()==LT_BINARY, "ROC evalution requires binary labels.");
+	REQUIRE(predicted, "No predicted labels provided.\n");
+	REQUIRE(ground_truth, "No ground truth labels provided.\n");
+	REQUIRE(
+	    predicted->get_label_type() == LT_BINARY,
+	    "Given predicted labels (%d) must be binary (%d).\n",
+	    predicted->get_label_type(), LT_BINARY);
+	REQUIRE(
+	    ground_truth->get_label_type() == LT_BINARY,
+	    "Given ground truth labels (%d) must be binary (%d).\n",
+	    ground_truth->get_label_type(), LT_BINARY);
 
 	return evaluate_roc((CBinaryLabels*)predicted,(CBinaryLabels*)ground_truth);
 }