tensorflow / tfx-addons

Developers helping developers. TFX-Addons is a collection of community projects to build new components, examples, libraries, and tools for TFX. The projects are organized under the auspices of the special interest group, SIG TFX-Addons. Join the group at http://goo.gle/tfx-addons-group

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Unified BulkInference for TFX Pipeline

ConfuzedCoder opened this issue · comments

TFX is framework neutral and can be used with other non-TF ML frameworks but the current version of the bulkInference only supports the Tensorflow model, which limits the ability of TFX. I would like to enhance the buikInference so that it can support many other non-TF frameworks like PyTorch and Xgboost.

Great idea. @tandelDipak do you have a suggestion for the component API?

It's just a thought right now, I don't have anything concrete, I will start working on it, and then we can have some discussion.