Import or view models with Machine Learning extension for Azure Data Studio (Preview)

Learn how to use the Machine Learning extension for Azure Data Studio to import an ONNX model or view already imported models in your database.

Important

Import and view in a database with the Machine Learning extension currently only supports Machine Learning Services in Azure SQL Managed Instance and Azure SQL Edge with ONNX.

Prerequisites

View models

Follow the steps below to view ONNX models that are stored in your database.

  1. Select Import or view models.

  2. If you're asked to install onnxruntime, mlflow, and mlflow-dbstore, select Yes.

  3. Select the Models database and Models table where your models are stored in.

This will show a list of your models. You can edit the model name and description, or delete a model from the list.

Import a new model

Follow the steps below to import an ONNX model in your database.

  1. Select Import or view models.

  2. If you're asked to install onnxruntime, mlflow, and mlflow-dbstore, select Yes.

  3. Select Import models.

  4. Select the Models database you want to store the imported model in.

  5. Select the Models table you want to store the imported model in. You can either choose an Existing table or create a New table. Select Next.

  6. Select where your model is located and Select Next. You can use:

    • File upload. Choose this to use a model from a file. Select the model file under Source files and Select Next.
    • Azure Machine Learning. Choose this to use a model from Azure Machine Learning. First, Sign in to Azure. Then select your Azure account, Azure subscription, Azure resource group, and Azure ML workspace. Select the model you want to use and Select Next.
  7. Enter the model Name and Description and Select Deploy to store the model in your database.

Note

The Machine Learning extension is currently in preview. Therefore, the table schema where the models are stored might change in the future.

Next steps