16 lines
393 B
C++
16 lines
393 B
C++
#include <gtest/gtest.h>
|
|
#include <xgboost/objective.h>
|
|
#include <string>
|
|
#include "../helpers.h"
|
|
|
|
namespace xgboost {
|
|
|
|
TEST(Plugin, ExampleObjective) {
|
|
xgboost::GenericParameter tparam = CreateEmptyGenericParam(GPUIDX);
|
|
auto * obj = xgboost::ObjFunction::Create("mylogistic", &tparam);
|
|
ASSERT_EQ(obj->DefaultEvalMetric(), std::string{"error"});
|
|
delete obj;
|
|
}
|
|
|
|
} // namespace xgboost
|