how to graphically display the tree?

[greghor]

Is there any tool available to visualize the chaid tree? Maybe something equivalent to export_graphiz for CART?

[Rambatino]

There is this: https://github.com/Rambatino/CHAID/blob/master/CHAID/tree.py#L198

Which uses https://github.com/Rambatino/CHAID/blob/master/CHAID/tree.py#L2

If you look at the treelib source code there is this: https://github.com/caesar0301/treelib/blob/master/treelib/tree.py#L865

Which I think is what you want. So you should be able to do:

tree.to_tree().to_graphviz(filename="") # untested

[greghor]

Thanks for your reply! I ll take a look at it!

[Rambatino]

@greghor have a look at https://github.com/Rambatino/CHAID/pull/94