AmirLyall
Databricks Employee
Databricks Employee

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