tests/cpp: Add tests for Metric RMSE

2016-12-04 13:15:09 +05:30
parent 582c373274
commit 03abd47f49
3 changed files with 32 additions and 0 deletions
--- a/tests/cpp/helpers.cc
+++ b/tests/cpp/helpers.cc
@@ -49,3 +49,14 @@ void CheckObjFunction(xgboost::ObjFunction * obj,
      << " weight=" << weights[i];
  }
 }
+
+xgboost::bst_float GetMetricEval(xgboost::Metric * metric,
+                                 std::vector<xgboost::bst_float> preds,
+                                 std::vector<xgboost::bst_float> labels,
+                                 std::vector<xgboost::bst_float> weights) {
+  xgboost::MetaInfo info;
+  info.num_row = labels.size();
+  info.labels = labels;
+  info.weights = weights;
+  return metric->Eval(preds, info, false);
+}