使用Optuna进行PyTorch模型的超参数调优
Optuna是一个开源的超参数优化框架,Optuna与框架无关,可以在任何机器学习或深度学习框架中使用它。本文将以表格数据为例,使用Optuna对PyTorch模型进行超参数调优。
Optuna可以使用python pip安装,如pip install Optuna。也可以使用conda install -c conda-forge Optuna,安装基于Anaconda的python发行版。
正如您所看到的,使用基本python语言的几行代码,您可以为任何神经网络创建并执行试验。
https://avoid.overfit.cn/post/86d01db7af8845b88ff1c753e7200a50