Skip to content

components model_prediction

github-actions[bot] edited this page Jun 13, 2023 · 32 revisions

Model Prediction

model_prediction

Overview

Description: Generate predictions on a given mlflow model for supported tasks.

Version: 0.0.9

Tags

type : evaluation sub_type : inference Preview

View in Studio: https://ml.azure.com/registries/azureml/components/model_prediction/version/0.0.9

Inputs

Name Description Type Default Optional Enum
task Task type string tabular-classification ['tabular-classification', 'tabular-classification-multilabel', 'tabular-regression', 'text-classification', 'text-classification-multilabel', 'text-named-entity-recognition', 'text-summarization', 'question-answering', 'text-translation', 'text-generation', 'fill-mask', 'image-classification', 'image-classification-multilabel']
test_data Test Data uri_file True
test_data_mltable Test Data MLTable path mltable True
mlflow_model Mlflow Model - registered model or output of a job with type mlflow_model in a pipeline mlflow_model
label_column_name Label column name in provided test dataset (Ex: label) string True
input_column_names Input column names in provided test dataset (Ex : column1). Add comma delimited values in case of multiple input columns (Ex : column1,column2) string True
evaluation_config Additional parameters required for evaluation. uri_file True
evaluation_config_params JSON Serielized string of evaluation_config string True

ADDED TO ACCOMMODATE CURRENT DESIGNER

Name Description Type Default Optional Enum
model_uri [DEPRECATED] The location, in URI format, of the MLflow model, for example: - runs:/<mlflow_run_id>/run-relative/path/to/model - models:/<model_name>/<model_version> - models:/<model_name>/<stage> string True
device string auto ['auto', 'cpu', 'gpu']
batch_size integer True

Outputs

Name Description Type
predictions uri_file
predictions_mltable mltable
prediction_probabilities uri_file
prediction_probabilities_mltable mltable
ground_truth uri_file
ground_truth_mltable mltable

Environment

azureml://registries/azureml/environments/model-evaluation/versions/8

Clone this wiki locally