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

Does TensorFlow Dececision Forests support batch training? #110

Closed wangbingnan136 closed 2 years ago

rstz commented 2 years ago

TF-DF always trains on the full dataset after it is read, hence the model quality will not vary based on the batch size. In other words, you can modify the batch size for IO speed (as is done for example in the model composition tutorial), but there is not need to tune the batch size. See the section Do not tune the batch size in the migration guide.