NNI (Neural Network Intelligence)简介

是什么

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

优势

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

NNI (Neural Network Intelligence)简介_第1张图片

安装

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

怎么用

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

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

NNI (Neural Network Intelligence)简介_第2张图片

相关网址

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

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

你可能感兴趣的:(工具,NNI,自动机器学习,超参调优,调参,工具)