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/metric/test_elementwise_metric.cc
+++ b/tests/cpp/metric/test_elementwise_metric.cc
@@ -0,0 +1,14 @@
+// Copyright by Contributors
+#include <xgboost/metric.h>
+
+#include "../helpers.h"
+
+TEST(Metric, RMSE) {
+  xgboost::Metric * metric = xgboost::Metric::Create("rmse");
+  ASSERT_STREQ(metric->Name(), "rmse");
+  EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}), 0, 1e-10);
+  EXPECT_NEAR(GetMetricEval(metric,
+                            {0.1, 0.9, 0.1, 0.9},
+                            {  0,   0,   1,   1}),
+              0.6403, 0.001);
+}