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.h
+++ b/tests/cpp/helpers.h
@@ -13,6 +13,7 @@

 #include <xgboost/base.h>
 #include <xgboost/objective.h>
+#include <xgboost/metric.h>

 std::string TempFileName();

@@ -29,4 +30,10 @@ void CheckObjFunction(xgboost::ObjFunction * obj,
                      std::vector<xgboost::bst_float> out_grad,
                      std::vector<xgboost::bst_float> out_hess);

+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 = std::vector<xgboost::bst_float> ());
+
 #endif