* Build plugin example with CMake.

* Remove plugin, cuda related code in automake & autoconf files.

* Fix typo in GPU doc.

2019-08-18 16:54:34 -04:00

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:

set(PLUGINS_SOURCES ${PLUGINS_SOURCES}
    ${xgboost_SOURCE_DIR}/plugin/example/custom_obj.cc PARENT_SCOPE)

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