When using dataset that is different from the training for trees models - does not draw

[StepanWorkV]

dtreeviz/dtreeviz/trees.py

Line 1255 in a0e85a0

# when using another dataset than the training dataset, some leaves could have 0 samples.

dtreeviz/dtreeviz/trees.py

Line 1257 in a0e85a0

if sum(counts) == 0:

Example:

Would work just fine

viz = DTreeVizAPI(
  ShadowDecTree.get_shadow_tree(
        decision_tree,
        X_train, <--- Train Data
        y_train,
        feature_names=X_train.columns.tolist(),
        target_name="target",
        class_names=labels,
    )
)

vw = viz.view(
    x=X_train.iloc[index],
    fancy=False,
    show_node_labels=False,
    show_just_path=False,
)

This would fail because of the issue described in the comment. Some nodes will not be generated, however, their file paths will be written into SVG file. When passing to graphviz later, this would error due to "file not found" for those missing plots.

viz = DTreeVizAPI(
  ShadowDecTree.get_shadow_tree(
        decision_tree,
        X_sample, <--- Some sample data that might not have all the leaves 
        y_sample,
        feature_names=X_sample.columns.tolist(),
        target_name="target",
        class_names=labels,
    )
)

vw = viz.view(
    x=X_sample.iloc[index],
    fancy=False,
    show_node_labels=False,
    show_just_path=False,
)

[tlapusan]

Hi @StepanWorkV, I do remember that this was solved. I have to look into the code/history