Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/data visualization Secret

Created October 10, 2020 08:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/332353acec6faab1a69c24f7e4834b1c to your computer and use it in GitHub Desktop.
Save amankharwal/332353acec6faab1a69c24f7e4834b1c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data visualization
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