Questions tagged [dtreeviz]

A Python library for decision tree visualization and model interpretation. Currently supports `scikit-learn`, `XGBoost`, and Spark `MLlib` trees.

A Python library for decision tree visualization and model interpretation. Currently supports scikit-learn, XGBoost, and Spark MLlib trees.

20 questions
0
votes
1 answer

Path error when visualising decision tree classification PySpark using dtreeviz

I am trying to visualise my decision tree classification using the code in GitHub in the following link https://github.com/parrt/dtreeviz/blob/master/notebooks/dtreeviz_spark_visualisations.ipynb when I am implementing the code: df =…
Roaa
  • 1
  • 3
0
votes
0 answers

Dtreeviz - Same feature on tree level

So I used dtreeviz to get a better view of my decision tree, I quadruple checked all the data, model and everything I did before plotting the decision tree, but when looking at it, we can see on the first level of the tree, that the two features are…
arlaine
  • 199
  • 8
0
votes
1 answer

Subtrees visualization in dtreeviz

I am new to dtreeviz. I am struggling with a very deep decision tree that is very difficult to visualize (overfitting is not an issue for my task). I would like to know if there is a way to visualize only some nodes of the three (e.g., first 5…
0
votes
1 answer

dtreeviz replace plot, regression too many points

How can I replace the node-plots from dtreeviz by a custom plot function from me? Alternatively: I want to replace the dtreeviz-plots with a 2d-histogram: y-axis=y-values, x-axis: values from the split feature, grid over the plot, each grid-cell…
groebsgr
  • 131
  • 1
  • 14
0
votes
0 answers

keep getting error "Input contains NaN, infinity or a value too large for dtype('float32')."

Trying to create tree for this data, not sure what is wrong exactly, but keep getting same error. import pandas as pd import seaborn as sns from sklearn.model_selection import train_test_split train, test =…
masteff
  • 1
  • 2
1
2