Add any required translation code to allow imodels trees to be plotted with dtreeviz. This basically boils down to successfully generating a ShadowDecTree object from an imodels tree.
We can reuse the existing ShadowSKDTreeconstructor by converting imodels trees into sklearn objects, then calling:
sk_dtree = ShadowSKDTree(tree_classifier, X, y, features, target, [0, 1])
Alternatively, we can make an imodels specific implementation of ShadowDecTree, similar to the sklearn implementation here, but that may be more work than necessary.
Add any required translation code to allow
imodels
trees to be plotted withdtreeviz
. This basically boils down to successfully generating aShadowDecTree
object from animodels
tree.We can reuse the existing
ShadowSKDTree
constructor by convertingimodels
trees intosklearn
objects, then calling:Alternatively, we can make an
imodels
specific implementation ofShadowDecTree
, similar to thesklearn
implementation here, but that may be more work than necessary.