tensorflow / lattice

Lattice methods in TensorFlow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Info: Is there any way to plot feature importance in lattice models?

as24-asingla opened this issue · comments

Hi,
We are using TF lattice model over a bunch of features. However, we want to understand if there is a way to plot feature importance in TF lattice?
Thanks in advance :)

There is no feature importance tool in the TFL codebase, but you can try partial dependence plots in the What-If tool as part of tensorboard in TFX, or standalone invocation in colab (see examples).

image
Generally speaking, the slope/variation of the PDP plot is a measure of feature importance.