components mlflow_model_local_validation - Azure/azureml-assets GitHub Wiki

MLFlow model local validation

mlflow_model_local_validation

Overview

Validates if a MLFLow model can be loaded on a compute and is usable for inferencing.

Version: 0.0.16

Tags

Preview

View in Studio: https://ml.azure.com/registries/azureml/components/mlflow_model_local_validation/version/0.0.16

Inputs

Name Description Type Default Optional Enum
model_path MLFlow model to be validated mlflow_model
test_data_path Test dataset for model inferencing uri_file True
column_rename_map Provide mapping of dataset column names that should be renamed before inferencing. eg: col1:ren1; col2:ren2; col3:ren3 string True
task_name A Hugging face task on which model was trained on string True ['chat-completion', 'fill-mask', 'token-classification', 'question-answering', 'summarization', 'text-generation', 'text2text-generation', 'text-classification', 'translation', 'image-classification', 'image-classification-multilabel', 'image-object-detection', 'image-instance-segmentation', 'image-to-text', 'text-to-image', 'text-to-image-inpainting', 'image-text-to-text', 'image-to-image', 'zero-shot-image-classification', 'mask-generation', 'video-multi-object-tracking', 'visual-question-answering']

Outputs

Name Description Type
mlflow_model_folder Validated input model. Here input model is used to block further steps in pipeline job if local validation fails uri_folder

Environment

azureml://registries/azureml/environments/python-sdk-v2/versions/23

⚠️ **GitHub.com Fallback** ⚠️