其他分享
首页 > 其他分享> > NNI (Neural Network Intelligence)简介

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.cn/zh-cn/news/features/nni

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

标签:zh,Network,nni,参调,Intelligence,NNI
来源: https://blog.csdn.net/wxq1999/article/details/117460377