ab357dd41c
* Build plugin example with CMake. * Remove plugin, cuda related code in automake & autoconf files. * Fix typo in GPU doc.
20 lines
708 B
Markdown
20 lines
708 B
Markdown
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](custom_obj.cc)
|
|
- Register this extension to xgboost via a registration macro
|
|
- In this example ```XGBOOST_REGISTER_OBJECTIVE``` in [this line](custom_obj.cc#L78)
|
|
- 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.
|
|
|
|
<!-- LocalWords: XGBoost
|
|
-->
|