利用NSGA-III算法优化机器学习模型
通过Optuna库实现机器学习模型超参数的优化与可视化,通过精心设计的目标函数,将搜索多个超参数空间,最终确定使模型性能最优的参数组合
为了更直观地展示调参过程,最后利用3D曲面图对调参效果进行可视化
本案例优化的是随机森林的超参数,优化算法采用NGSA-III,优化的算法和预测模型可以根据要求修改,如NSGA-II,MOEA/D
一键运行代码生成的图片包括:光滑3D曲面图展示RMSE变化(图1)
相关系数气泡热力图(图2)
训练集与测试集预测值对比及置信区间可视化(图3)
ID:5648
详询客服 微信shujuqudong1 或shujuqudong6 或 qq68823886 或 27699885
图文详情请查看: http://matup.cn/862370914355.html