cancel
Showing results for 
Search instead for 
Did you mean: 

Is there a way to import .tf or .keras models?

AutoDR
DataRobot Employee
DataRobot Employee

Is there a way to import .tf or .keras models?

Robot 1:

I have a customer who wants to import these model types.


Robot 2:

We don’t allow any form of model import (pmml, tf, keras, h5, json etc.), except for custom models.


Robot 2:

That said, you can use custom models to do whatever you want, including importing whatever you wish.


Robot 1:

I think he is trying to do Custom Inference. Can he use this or only .h5? I don't really understand the tradeoff of one version of the model objects versus another. I've only used h5 and JSON. I think he was also curious if we have any support for importing JSON weights.


Robot 2:

We do not support importing model files, except for custom models--he'll need to write a custom inference model to load the file and score data.


Robot 2:

But yes, we support custom inference models. You can do literally whatever you want in custom inference models.


Robot 1:

Thanks, sorry if I'm being dense / confused - so with Custom Inference he should be able to load the .pb, .tf,.keras files?


Robot 2:

Yes— he will need to write his own Python code. So if he can write a Python script to load the .pd, .tf, .keras, or .whatever file and score data with it, he can make that script a custom inference model.


Robot 1:

Ohhh of course 🙂 now i understand. Duh, Robot 1. Thanks!

0 Kudos
0 Replies