NNI (Neural Network Intelligence)简介

是什么

NNI (Neural Network Intelligence) 是一个轻量但强大的自动机器学习（AutoML）工具包，能帮助用户自动地进行特征工程、神经网络架构搜索、超参调优以及模型压缩。

优势

支持多框架、多训练平台，中文文档，易用的命令行工具和web界面

安装

pip install nni 或 python3 -m pip install --upgrade nni

怎么用

以超参优化为例。首先，定义好需要搜索的超参空间；然后，在需要调参的网络启动之前，通过 NNI 的接口读取参数并在训练中将精确度等指标传入 NNI；最后，配置好要使用的调参算法等，即可开始。

NNI 的超参调优不仅能用于机器学习，对于各类系统、数据库的繁杂参数都可以根据实际场景进行有针对性的调优。使用过程和超参调优非常类似，通过 Python 为系统传入不同的参数配置，然后将确定的调优指标（如读写速度，磁盘空间大小等）回调给 NNI 即可。

相关网址

ms的简介：https://www.msra/zh-cn/news/features/nni

项目地址：https://github/microsoft/nni/blob/master/README_zh_CN.md