Multiple outputs with TFDF

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

Multiple outputs with TFDF #119

Closed ghost closed 2 years ago

ghost commented 2 years ago

Hi,

I have a basic question about multi-output regression and multi-label classification using TFDF package. Does TFDF support such cases? If yes, are there some examples on how to implement such models with TFDF and some caveat to be mindful of?

Thanks.

mrsergazinov commented 2 years ago

Hey! It appears this capability is not supported. See issue here: https://github.com/tensorflow/decision-forests/issues/19

ghost commented 2 years ago

@mrsergazinov Thanks for the pointer. Appreciate!