NNI (Neural Network Intelligence) is a toolkit to help users run automated machine learning experiments. The tool dispatches and runs trial jobs that generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different environments (e.g. local machine, remote servers and cloud).
AutoML experiment Training Services
┌────────┐ ┌────────────────────────┐ ┌────────────────┐
│ nnictl │ ─────> │ nni_manager │ │ Local Machine │
└────────┘ │ sdk/tuner │ └────────────────┘
│ hyperopt_tuner │
│ evolution_tuner │ trial jobs ┌────────────────┐
│ ... │ ────────> │ Remote Servers │
├────────────────────────┤ └────────────────┘
│ trial job source code │
│ sdk/annotation │ ┌────────────────┐
├────────────────────────┤ │ Yarn,K8s, │
│ nni_board │ │ ... │
└────────────────────────┘ └────────────────┘
- You want to try different AutoML algorithms for your training code (model) at local
- You want to run AutoML trial jobs in different environments to speed up search (e.g. remote servers and cloud)
- As a researcher and data scientist, you want to implement your own AutoML algorithms and compare with other algorithms
- As a ML platform owner, you want to support AutoML in your platform
pip Installation Prerequisites
- linux (ubuntu 16.04 or newer version has been well tested)
- python >= 3.5
- git, wget
python3 -m pip install -v --user git+https://github.com/Microsoft/nni.git@v0.1
source ~/.bashrc
It only requires 3 steps to start an experiment on NNI:
NNI provides a set of examples in the package to get you familiar with the above process. In the following example [/examples/trials/mnist], we had already set up the configuration and updated the training codes for you. You can directly run the following command to start an experiment.
NOTE: The following example is an experiment built on TensorFlow, make sure you have TensorFlow installed before running the following command.
Try it out:
nnictl create --config ~/nni/examples/trials/mnist/config.yml
In the command output, find out the Web UI url and open it in your browser. You can analyze your experiment through WebUI, or browse trials' tensorboard.
To learn more about how this example was constructed and how to analyze the experiment results in NNI Web UI, please refer to How to write a trial run on NNI (MNIST as an example)?
Please refer to Get Started Tutorial for more detailed information.
- Tutorial of NNI python annotation.
- Tuners supported by NNI.
- How to enable early stop (i.e. assessor) in an experiment?
- How to run an experiment on multiple machines?
- How to write a customized tuner?
- How to write a customized assessor?
- How to resume an experiment?
- Tutorial of the command tool nnictl.
This project welcomes contributions and suggestions, we are constructing the contribution guidelines, stay tuned =).
We use GitHub issues for tracking requests and bugs.