How to I convert a custom sklearn transformer that uses a Tensorflow embedding model? #5695
Unanswered
shezadkhan137
asked this question in
Q&A
Replies: 1 comment
-
You may create a custom converter mapped to your TextEmbeddingTransformer (see http://onnx.ai/sklearn-onnx/auto_tutorial/plot_icustom_converter.html) and append the onnx graph coming from tf2onnx. A full example is implemented in one unit test https://github.com/onnx/sklearn-onnx/blob/main/tests/test_other_converter_library_pipelines.py. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi, I have a sklearn pipeline that uses a transformer that calls a tensorflow embedding model (USEv4):
I then use it in a pipeline:
I'm struggling to figure out how to convert the whole pipeline to onnx. I know I can convert the embedding model itself to onnx, but I'm not sure what to do from then.
Beta Was this translation helpful? Give feedback.
All reactions