cf2d86a4f6
* Add travis sanitizers tests. * Add gcc-7 in Travis. * Add SANITIZER_PATH for CMake. * Enable sanitizer tests in Travis. * Fix memory leaks in tests. * Fix all memory leaks reported by Address Sanitizer. * tests/cpp/helpers.h/CreateDMatrix now returns raw pointer.
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
// 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.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f},
|
|
{0, 1, 2}),
|
|
0.666f, 0.001f);
|
|
|
|
delete metric;
|
|
}
|
|
|
|
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.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f, 0.1f},
|
|
{0, 1, 2}),
|
|
2.302f, 0.001f);
|
|
|
|
delete metric;
|
|
}
|