search

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

GPU support #37

Closed Howard-ll closed 3 years ago

Howard-ll commented 3 years ago

I just wanted to know if there is a plan for GPU. My codes are in tensorflow. Therefore, running on GPU is very important in terms of throughput.

janpfeifer commented 3 years ago

There is no concrete plans yet: we do have some experimental implementation and it does help. But not by as much as you may expect, since decision forests is not so much about number crunching.

We would like to work on that, but it's lower on our priority list, so no ETA we can give.

Feel free to open an enhancement type of request, we like those to keep track, and help gauge the interest -- which in turn guides our prioritization.