search

tensorflow / lattice

Lattice methods in TensorFlow
Apache License 2.0
518 stars 94 forks source link

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

Closed as24-asingla closed 3 years ago

as24-asingla commented 3 years ago

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 :)

mmilanifard commented 3 years ago

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.