This project implements a method that converts a trained XGBoost into a single decision tree. At this stage, this package is ideal for binary classification challenges. It can be applied for challenges with a few classes (2-3) but is not recommended for numerous classes, at least for the current implementation. Please see the included .ipynb file for further explanations of the API.
-
Notifications
You must be signed in to change notification settings - Fork 6
sagyome/XGBoostTreeApproximator
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
This project implements a method that converts a trained XGBoost classification tree into a single decision tree.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published