I'm trying to visualize the result of by CatBoostClassifier in Databricks. I have graphviz ==0.18.2 installed on my cluster.
When I use the following code (NB. model is my trained CatBoostClassifier):
model.plot_tree(tree_idx=0)
I don't get a tree, but I get the following output: <graphviz.graphs.Digraph at 0x7f86330d9640>.
This doesn't seem to happen in this post.
Any suggestions on how I can visualize my tree?