xgboost/plugin/example

XGBoost Plugin Example

This folder provides an example of xgboost plugin.

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

• Create your source .cc file, implement a new extension
  • In this example custom_obj.cc
• Register this extension to xgboost via registration macr
  • In this example XGBOOST_REGISTER_OBJECTIVE in this line
• Create a plugin.mk on this folder

To add this plugin, add the following line to config.mk(template in make/config.mk).

# Add plugin by include the plugin in config
include plugin/example/plugin.mk

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