History

* Add a new ctor to tensor for `initilizer_list`.
* Change labels from host device vector to tensor.
* Rename the field from `labels_` to `labels` since it's a public member.

2021-12-17 00:58:35 +08:00

custom_obj.cc

Convert labels into tensor. (#7456 )

2021-12-17 00:58:35 +08:00

README.md

Update plugin instructions for CMake build (#6289 )

2020-10-26 17:42:07 -07:00

README.md

XGBoost Plugin Example

This folder provides an example of implementing xgboost plugin.

There are three steps you need to do to add a plugin to xgboost

Create your source .cc file, implement a new extension
- In this example custom_obj.cc
Register this extension to xgboost via a registration macro
- In this example XGBOOST_REGISTER_OBJECTIVE in this line
Add a line to xgboost/plugin/CMakeLists.txt:

target_sources(objxgboost PRIVATE ${xgboost_SOURCE_DIR}/plugin/example/custom_obj.cc)

Then you can test this plugin by using objective=mylogistic parameter.