cancel
Showing results for 
Search instead for 
Did you mean: 

Custom Objects in Keras Model

Eric Yang
Blue LED

Custom Objects in Keras Model

For one of my keras networks, I'm using a custom loss. Normally when I load the model I have to do something like: 

model = load_model(modelFile, custom_objects={ 'loss': custom_loss_func})

 Where the modelFile is the model that's saved as an h5 file. How would I deploy this type of model on DataRobot?

Labels (1)
0 Kudos
1 Reply
Kyle Dettman
Data Scientist
Data Scientist

If you want to use this model as part of a blueprint for the purposes of model building, you can import it as a custom estimator task and modify a potentially existing Keras blueprint by swapping this in.

 

If this model is one that you already have trained and you want to monitor it in MLOps, you can import it as a custom model.

 

For both situations, the "load_model" function would go in a specific file called "custom.py".

0 Kudos