Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tests/cpp: Add test for elementwise_metric.cc

Browse Source
This commit is contained in:
AbdealiJK 2016-12-04 13:39:00 +05:30 committed by Tianqi Chen
parent 03abd47f49
commit 4a2ef130a7
1 changed files with 55 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
tests/cpp/metric/test_elementwise_metric.cc
View File

@ -12,3 +12,58 @@ TEST(Metric, RMSE) {
{ 0, 0, 1, 1}),
0.6403, 0.001);
}
TEST(Metric, MAE) {
xgboost::Metric * metric = xgboost::Metric::Create("mae");
ASSERT_STREQ(metric->Name(), "mae");
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.5, 0.001);
}
TEST(Metric, LogLoss) {
xgboost::Metric * metric = xgboost::Metric::Create("logloss");
ASSERT_STREQ(metric->Name(), "logloss");
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}),
1.2039, 0.001);
}
TEST(Metric, Error) {
xgboost::Metric * metric = xgboost::Metric::Create("error");
ASSERT_STREQ(metric->Name(), "error");
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.5, 0.001);
EXPECT_ANY_THROW(xgboost::Metric::Create("error@abc"));
delete metric;
metric = xgboost::Metric::Create("error@0.5");
EXPECT_STREQ(metric->Name(), "error");
delete metric;
metric = xgboost::Metric::Create("error@0.1");
ASSERT_STREQ(metric->Name(), "error@0.1");
EXPECT_STREQ(metric->Name(), "error@0.1");
EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}), 0, 1e-10);
EXPECT_NEAR(GetMetricEval(metric,
{0.1, 0.2, 0.1, 0.2},
{ 0, 0, 1, 1}),
0.5, 0.001);
}
TEST(Metric, PoissionNegLogLik) {
xgboost::Metric * metric = xgboost::Metric::Create("poisson-nloglik");
ASSERT_STREQ(metric->Name(), "poisson-nloglik");
EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}), 0.5, 1e-10);
EXPECT_NEAR(GetMetricEval(metric,
{0.1, 0.2, 0.1, 0.2},
{ 0, 0, 1, 1}),
1.1280, 0.001);
}