Perform Automated Machine Learning for Free with Microsoft’s Open Source Python Toolkit

Pranav Dar 07 May, 2019 • 2 min read


  • Neural Network Intelligence (NNI) is Microsoft’s open-source toolkit for automated machine learning
  • It helps you perform ML tasks like hyperparameter tuning and neural architecture search
  • You need to have Python 3.5 or greater to use NNI



When you hear the words “automated machine learning”, what comes to your mind first? For me, it’s usually’s Driverless AI, or Google’s Cloud AutoML. Microsoft is probably a bit down in that list (Azure, anyone?).

But that list might be about to see some changes. Microsoft has released an open-source automated machine learning toolkit on GitHub that helps a user perform neural architecture search and hyperparameter tuning. Microsoft is calling the toolkit ‘Neural Network Intelligence (NNI)’.

According to Microsoft, “the tool dispatches and runs trial jobs that generated are by tuning algorithms to search for the best neural architecture and/or hyper-parameters at different environments (e.g. local, remote servers and cloud)”. The below diagram illustrates this point well:

So who does this NNI toolkit target? And why should you consider using it (or at least giving it a go)? Use the below checklist to find out:

  • You want to support AutoML in your existing machine learning infrastructure
  • You want to implement your own AutoML algorithms and compare them with other existing algorithms (perhaps benchmarks)
  • You want to try out different versions of AutoML algorithms

You can install NNI through pip by using the below command:

pip3 install -v --user git+[email protected]
source ~/.bashrc

Note that you’ll need to have Python version 3.5 or greater to use this toolkit.


Our take on this

I’ll be honest – when I took my first steps into the dreamy world of data science, I hadn’t imagined autoML picking up so quickly. I used to hear my seniors talk about how all of that was at least 5-7 years away (back in 2016). Just goes to show how quickly technology has advanced, it’s even taken people in-the-know by surprise.

And we’re already at the stage of seeing autoML going open-source (or at least parts of it)! I covered Auto-Keras last month and that was quite a big deal in the ML community. I’m sure Microsoft’s NNI will help speed up the automated designing of models to quite an extent as well. It’s definitely worth exploring and since it’s free, there are no excuses for not doing so!


Subscribe to AVBytes here to get regular data science, machine learning and AI updates in your inbox!


Pranav Dar 07 May 2019

Senior Editor at Analytics Vidhya. Data visualization practitioner who loves reading and delving deeper into the data science and machine learning arts. Always looking for new ways to improve processes using ML and AI.

Frequently Asked Questions

Lorem ipsum dolor sit amet, consectetur adipiscing elit,

Responses From Readers