search

tensorflow / decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.
Apache License 2.0
663 stars 110 forks source link

model_plotter.plot_model_in_colab has No response #183

Closed 983183947 closed 1 year ago

983183947 commented 1 year ago

I use model_plotter.plot_model_in_colab has no response（not show image）

rstz commented 1 year ago

Hi, can you please share a reproduction notebook on https://colab.research.google.com/ ?

983183947 commented 1 year ago

code in colab is ok。but when i use tensorflow with docker it not show image。

rstz commented 1 year ago

Hi, this method only works in Colab, so this is WIP.

You can get the HTML representation of the model with model_plotter. plot_model(model) and view that with any html viewer you want, or you can get a text representation of the individual trees via inspector.extract_tree(tree_idx=IDX_OF_THE_TREE_YOU_CARE_ABOUT).pretty()