《表2 不同学习率的模型预测实验结果》
由实验结果(表1、表2)可知不同的学习率各有优缺点:学习率较大时,模型训练速度较快,不容易收敛且实验结果较差;当学习率为0.1时,最好的测试结果为RMSE=4.665 762,远不及学习率为0.001的测试结果,且训练集的损失值相当大,表明ATCN模型在使用较大或较小学习率时存在一定问题;当学习率为0.01时,无论是训练集还是测试集,效果都呈现相似的收敛趋势,测试结果最终收敛在3.540 056.因此,α=0.01应为最优的学习率参数.
图表编号 | XD00160900700 严禁用于非法目的 |
---|---|
绘制时间 | 2020.08.25 |
作者 | 廖恩红、舒娜、李加伟、庞雄文 |
绘制单位 | 广东工程职业技术学院信息工程学院、华南师范大学计算机学院、华南师范大学计算机学院、华南师范大学计算机学院 |
更多格式 | 高清、无水印(增值服务) |