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

enhancement request: tfdf.keras.models.load_model #66

Closed bpatay closed 3 years ago

bpatay commented 3 years ago

Having difficulty using saved models for inference.

Could you create a method that loads a saved TFDF model similar to tf.keras.models.load_model?