You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Add support for model signature and examples in dataset MlflowModelTrackingDataset
Context
MLflows support the addition of signature and examples in the models, which includes useful information in the model artifact view
Possible Implementation
At the moment I am passing the signature as a dictionary in the save_args, and using an aditional omegaconf resolver where I am transforming the dictionary to a ModelSignature object. Something similar could be achieved inside MlflowModelTrackingDataset.
Another alternative could be using mlflow infer_signature, however, I am not sure how can you pass the object to infer the schema from. The same thing happens with the example, which according to the documentation could be any of pandas.core.frame.DataFrame, numpy.ndarray, dict, list, csr_matrix, csc_matrix, str, bytes, tuple. At the moment I can pass just a dict or list in the catalog yml.
The text was updated successfully, but these errors were encountered:
Description
Add support for model signature and examples in dataset
MlflowModelTrackingDataset
Context
MLflows support the addition of signature and examples in the models, which includes useful information in the model artifact view
Possible Implementation
At the moment I am passing the signature as a dictionary in the save_args, and using an aditional omegaconf resolver where I am transforming the dictionary to a
ModelSignature
object. Something similar could be achieved insideMlflowModelTrackingDataset
.Another alternative could be using mlflow
infer_signature
, however, I am not sure how can you pass the object to infer the schema from. The same thing happens with the example, which according to the documentation could be any ofpandas.core.frame.DataFrame, numpy.ndarray, dict, list, csr_matrix, csc_matrix, str, bytes, tuple
. At the moment I can pass just a dict or list in the catalog yml.The text was updated successfully, but these errors were encountered: