plot smoothed tree

[markusloecher]

Is there any way to plot a shrunk classification tree fitted with your package ?
Since the object inherits from DecisionTreeClassifier I had hoped that tree.plot_tree works but that does not seem to be the case:

clf_datasets = [
    ("breast-cancer", "breast_cancer", "imodels")
]
X, y, feature_names = get_clean_dataset('breast_cancer', data_source='imodels')

# train-test split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20)

clf_bbs = ShrinkageClassifier(DecisionTreeClassifier(min_samples_leaf=25),shrink_mode="beta", 
alpha=10, beta=10)
clf_bbs.fit(X_train, y_train)

fig = plt.figure(figsize=(25,20))

_ = tree.plot_tree(clf_bbs, 
                   feature_names=feature_names,  
                   filled=True,proportion=True)