History

- Use the `linalg::Matrix` for storing gradients.
- New API for the custom objective.
- Custom objective for multi-class/multi-target is now required to return the correct shape.
- Custom objective for Python can accept arrays with any strides. (row-major, column-major)

2023-08-24 05:29:52 +08:00

custom_obj.cc

Use matrix for gradient. (#9508 )

2023-08-24 05:29:52 +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.