是什么
NNI (Neural Network Intelligence) 是一个轻量但强大的自动机器学习(AutoML)工具包,能帮助用户自动的进行特征工程,神经网络架构搜索,超参调优以及模型压缩。
优势
支持多框架、多训练平台,中文文档,易用的命令行工具和web界面
安装
pip install nni 或 python3 -m pip install --upgrade nni
怎么用
以超参优化为例。首先,定义好需要搜索的超参空间;然后,在需要调参的网络启动之前,通过 NNI 的接口读取参数并在训练中将精确度等指标传入 NNI;最后,配置好要使用的调参算法等,即可开始。
NNI 的超参调优不仅能用于机器学习,对于各类系统、数据库的繁杂参数都可以根据实际场景进行有针对性的调优。使用过程和超参调优非常类似,通过 Python 为系统传入不同的参数配置,然后将确定的调优指标(如读写速度,磁盘空间大小等)回调给 NNI 即可。
相关网址
ms的简介:https://www.msra.cn/zh-cn/news/features/nni
项目地址:https://github.com/microsoft/nni/blob/master/README_zh_CN.md