14 lines
365 B
C++
14 lines
365 B
C++
#include <gtest/gtest.h>
|
|
#include <xgboost/objective.h>
|
|
#include <string>
|
|
#include "../helpers.h"
|
|
|
|
namespace xgboost {
|
|
TEST(Plugin, ExampleObjective) {
|
|
xgboost::Context ctx = MakeCUDACtx(GPUIDX);
|
|
auto* obj = xgboost::ObjFunction::Create("mylogistic", &ctx);
|
|
ASSERT_EQ(obj->DefaultEvalMetric(), std::string{"logloss"});
|
|
delete obj;
|
|
}
|
|
} // namespace xgboost
|