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

tests/cpp: Add tests for multiclass_metric.cc

Browse Source
This commit is contained in:
AbdealiJK 2016-12-04 16:31:12 +05:30 committed by Tianqi Chen
parent a7e20555a3
commit 47ba2de7d4
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/cpp/objective/test_multiclass_metric.cc Normal file
View File

@ -0,0 +1,28 @@
// Copyright by Contributors
#include <xgboost/metric.h>
#include "../helpers.h"
TEST(Metric, MultiClassError) {
xgboost::Metric * metric = xgboost::Metric::Create("merror");
ASSERT_STREQ(metric->Name(), "merror");
EXPECT_ANY_THROW(GetMetricEval(metric, {0}, {0, 0}));
EXPECT_NEAR(GetMetricEval(
metric, {1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 1, 2}), 0, 1e-10);
EXPECT_NEAR(GetMetricEval(metric,
{0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1},
{0, 1, 2}),
0.666, 0.001);
}
TEST(Metric, MultiClassLogLoss) {
xgboost::Metric * metric = xgboost::Metric::Create("mlogloss");
ASSERT_STREQ(metric->Name(), "mlogloss");
EXPECT_ANY_THROW(GetMetricEval(metric, {0}, {0, 0}));
EXPECT_NEAR(GetMetricEval(
metric, {1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 1, 2}), 0, 1e-10);
EXPECT_NEAR(GetMetricEval(metric,
{0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1},
{0, 1, 2}),
2.302, 0.001);
}