Skip to content

Visualizing an XGBoost model in R using a sunburst plot (using inTrees)

Notifications You must be signed in to change notification settings

SidJain1412/XGBoostExplainability

Repository files navigation

XGBoostExplainability

An XGBoost (or any other ensemble model) is a black box, and one can't decipher why the predictions are what they are.

(view the Notebook.md file for the notebook, .Rmd files for the source code)

In this R notebook I demonstrate how to use the inTrees library to interpret and uncover the basic rules influencing the decision trees.

(or atleast my understanding of it)

Flow of inTrees:

Flow Diagram

Reference Paper

Documentation

Using the Mushroom dataset, classifying mushrooms into poisonous or edible based on various features (Dataset Link)

Releases

No releases published

Packages

No packages published

Languages