File: test_example_objective.cc

package info (click to toggle)
xgboost 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 13,796 kB
  • sloc: cpp: 67,502; python: 35,503; java: 4,676; ansic: 1,426; sh: 1,320; xml: 1,197; makefile: 204; javascript: 19
file content (13 lines) | stat: -rw-r--r-- 365 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
 
#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