cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

How to add signature to model logged through feature store?

vives
New Contributor II

It seems that the current method log_model from the FeatureStoreClient class lacks a way to pass in the model signature (as opposed as doing it through mlflow directly). Is there a workaround to append this information? Thanks!

3 REPLIES 3

Arnold69
New Contributor II

How to add signature to model logged through feature store? It seems that the current method log_model from the FeatureStoreClient class.

AmirLyall
New Contributor II

Hello!

You can log a model with a signature by passing a signature object as an argument with your log_model call. Please see here.

Here's an example of this in action in a databricks notebook.

Hope that helps!

-Amir

vives
New Contributor II

Hi Amir,

thanks for the reply. However, the issue arises when logging the model through the FeatureStoreClient class' method "log_model" (as seen in page 6 here https://docs.gcp.databricks.com/_static/documents/feature-store-python-api-reference-0-3-6.pdf), let me know if that helps!