Split-Up: dtreeviz (Part 1)

h1ros

Apr 15, 2019, 10:27:13 PM

Goal¶

This post aims to go through each function in dtreeviz module to fully understand what is implemented. After fully understanding this, I would like to contribute to this module and submit a pull request.

I really like this module and would like to see this works for other tree-based modules like XGBoost or Lightgbm. I found the exact same issue (issues 15) in github so I hope I could contribute to this issue.

You would just have to get ShadowDecisionTree wrappers for those trees.

Based on this comment, I need first understand the class object ShadowDecisionTree

Understand folder structure¶

In this post, we will deep dive into the core module dtreeviz

This module comprises of 4 python files.

__init__.py is empty so we can skip it.

Let's see one by one.