70 lines
2.4 KiB
C++
70 lines
2.4 KiB
C++
// 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);
|
|
}
|
|
|
|
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);
|
|
}
|