predict()
takes one argument, which is the model input data and returns the model output. The model input can be of arbitrary type as defined in DeployedModel.predict()
, but it must be serializable to json.log_model()
method, such as an ExperimentRun
, automatically supports logging custom models. They will be serialized in the right way for consumption downstream.RegisteredModelVersion
: