New decision forests library issue

Hi,

The new TF-DF library is quite impressive! However, it seems that there is a problem with plotting the tree. The following code does not work outside of Colab (for example Kaggle Kernels)?

tfdf.model_plotter.plot_model(model)

maybe there’s some configuration issue with D3 (the JS library to display this kind of visualization)

@Mathieu , I tried changing the imports and playing with the JS usage but I couldn’t figure it out what is the issue.

Hi,

Thanks @nncv for reporting the issue and thanks @lgusm for looking it :slight_smile: .

The method tfdf.model_plotter.plot_model is expected to be independent of colab, but there is probably an issue. I’ll take a look and comeback to you.

ETA: Tomorow.

1 Like