29 lines
1.2 KiB
C++
29 lines
1.2 KiB
C++
// Copyright by Contributors
|
|
#include <xgboost/objective.h>
|
|
#include <xgboost/context.h>
|
|
#include <limits>
|
|
|
|
#include "../helpers.h"
|
|
namespace xgboost {
|
|
TEST(Objective, DeclareUnifiedTest(HingeObj)) {
|
|
Context ctx = MakeCUDACtx(GetGPUId());
|
|
std::unique_ptr<ObjFunction> obj{ObjFunction::Create("binary:hinge", &ctx)};
|
|
|
|
float eps = std::numeric_limits<xgboost::bst_float>::min();
|
|
CheckObjFunction(obj,
|
|
{-1.0f, -0.5f, 0.5f, 1.0f, -1.0f, -0.5f, 0.5f, 1.0f},
|
|
{ 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 1.0f, 1.0f, 1.0f},
|
|
{ 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f},
|
|
{ 0.0f, 1.0f, 1.0f, 1.0f, -1.0f, -1.0f, -1.0f, 0.0f},
|
|
{ eps, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, eps });
|
|
CheckObjFunction(obj,
|
|
{-1.0f, -0.5f, 0.5f, 1.0f, -1.0f, -0.5f, 0.5f, 1.0f},
|
|
{ 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 1.0f, 1.0f, 1.0f},
|
|
{}, // Empty weight.
|
|
{ 0.0f, 1.0f, 1.0f, 1.0f, -1.0f, -1.0f, -1.0f, 0.0f},
|
|
{ eps, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, eps });
|
|
|
|
ASSERT_NO_THROW(obj->DefaultEvalMetric());
|
|
}
|
|
} // namespace xgboost
|