TPOT model in google ml-engine

121 Views Asked by At

I have a trained model using TPOT. When I tried loading the model to ml-engine it says: No module named tpot.builtins.stacking_estimator

The error makes sense since TPOT is an external package, not included in Cloud ML Engine runtime versions. Is there any way to get around it?

1

There are 1 best solutions below

2
Héctor Neri On

For what I can see, you only need PyPI to add the related dependencies. Checking the dependencies for the installation of TPOT, all of those can be installed through pip.

So, follow the documentation for "Adding standard (PyPI) dependencies" and include this module and all the related dependencies in your setup.py file. And follow the related steps to package and upload your training application.