Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created Oct 10, 2020
Embed
What would you like to do?
from dtreeviz.trees import *
viz = dtreeviz(clf,
X_train,
y_train.values,
target_name='Genre',
feature_names=list(X.columns),
class_names=list(labels.unique()),
histtype='bar',
orientation ='TD')
viz
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment