Get leafs values in class inherits from tensorflow.keras.Model

Hi All,

I would like to be able to get the leaves values when training ensemble that uses tfdf model as a layer, in custom class I created.

Calling tfdf(inputs) works just fine, but when trying to use tfdf.call_get_leaves(inputs) I got a tensor with shape None, None that I can’t use.

I reproduces the example here:

my_ensemble

Will appreciate your help, thank you!