Use Microsoft Cognitive Toolkit deep learning model with Azure HDInsight Spark cluster

In this article, you do the following steps.

  1. Run a custom script to install Microsoft Cognitive Toolkit on an Azure HDInsight Spark cluster.

  2. Upload a Jupyter notebook to the Spark cluster to see how to apply a trained Microsoft Cognitive Toolkit deep learning model to files in an Azure Blob Storage Account using the Spark Python API (PySpark)


How does this solution flow?

This solution is divided between this article and a Jupyter notebook that you upload as part of this tutorial. In this article, you complete the following steps:

  • Run a script action on an HDInsight Spark cluster to install Microsoft Cognitive Toolkit and Python packages.
  • Upload the Jupyter notebook that runs the solution to the HDInsight Spark cluster.

The following remaining steps are covered in the Jupyter notebook.

  • Load sample images into a Spark Resiliant Distributed Dataset or RDD
    • Load modules and define presets
    • Download the dataset locally on the Spark cluster
    • Convert the dataset into an RDD
  • Score the images using a trained Cognitive Toolkit model
    • Download the trained Cognitive Toolkit model to the Spark cluster
    • Define functions to be used by worker nodes
    • Score the images on worker nodes
    • Evaluate model accuracy

Install Microsoft Cognitive Toolkit

You can install Microsoft Cognitive Toolkit on a Spark cluster using script action. Script action uses custom scripts to install components on the cluster that are not available by default. You can use the custom script from the Azure Portal, by using HDInsight .NET SDK, or by using Azure PowerShell. You can also use the script to install the toolkit either as part of cluster creation, or after the cluster is up and running.

In this article, we use the portal to install the toolkit, after the cluster has been created. For other ways to run the custom script, see Customize HDInsight clusters using Script Action.

Using the Azure Portal

For instructions on how to use the Azure Portal to run script action, see Customize HDInsight clusters using Script Action. Make sure you provide the following inputs to install Microsoft Cognitive Toolkit.

  • Provide a value for the script action name.

  • For Bash script URI, enter

  • Make sure you run the script only on the head and worker nodes and clear all the other checkboxes.

  • Click Create.

Upload the Jupyter notebook to Azure HDInsight Spark cluster

To use the Microsoft Cognitive Toolkit with the Azure HDInsight Spark cluster, you must load the Jupyter notebook CNTK_model_scoring_on_Spark_walkthrough.ipynb to the Azure HDInsight Spark cluster. This notebook is available on GitHub at

  1. Clone the GitHub repository For instructions to clone, see Cloning a repository.

  2. From the Azure Portal, open the Spark cluster blade that you already provisioned, click Cluster Dashboard, and then click Jupyter notebook.

    You can also launch the Jupyter notebook by going to the URL https://<clustername> Replace <clustername> with the name of your HDInsight cluster.

  3. From the Jupyter notebook, click Upload in the top-right corner and then navigate to the location where you cloned the GitHub repository.

    Upload Jupyter notebook to Azure HDInsight Spark cluster

  4. Click Upload again.

  5. After the notebook is uploaded, click the name of the notebook and then follow the instructions in the notebook itself on how to load the data set and perform the tutorial.

See also


Create and run applications

Tools and extensions

Manage resources