3e26107a9c
* Rename `GenericParameter` to `Context`. * Rename header file to reflect the change. * Rename all references.
30 lines
1.2 KiB
C++
30 lines
1.2 KiB
C++
// Copyright by Contributors
|
|
#include <xgboost/objective.h>
|
|
#include <xgboost/context.h>
|
|
#include <limits>
|
|
|
|
#include "../helpers.h"
|
|
|
|
TEST(Objective, DeclareUnifiedTest(HingeObj)) {
|
|
xgboost::Context ctx = xgboost::CreateEmptyGenericParam(GPUIDX);
|
|
std::unique_ptr<xgboost::ObjFunction> obj {
|
|
xgboost::ObjFunction::Create("binary:hinge", &ctx)
|
|
};
|
|
|
|
xgboost::bst_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());
|
|
}
|