h2oai/driverlessai-recipes

driverless h2oai text_embedding_similarity_transformers.py doesnt show up

cyrilthank opened this issue · 5 comments

I am using the 21 day trial driverless h2oai

I am trying to upload custom recipe (text_embedding_similarity_transformers.py) for transform

after the save it doesnt show up as mentioned in the blog for similar use case

https://www.h2o.ai/blog/custom-machine-learning-recipes-the-ingredients-for-success/

can you please advise what i am doing wrong as i am trying to ascertain whether the bert embedding is used or not

HI @cyrilthank - Currently this transformer is disabled by default in the code. This is because of the GPU memory issue mentioned in #6. Thank you.

Thanks a lot @SudalaiRajkumar for the clarification. Thank you

Cyril

You could enable the recipe in code and use it but you may face some GPU memory issues.