《表3 模型超参数寻优范围及结果》
将模型的网络结构设定为1个输入层(19个结点)、1个输出层(3个结点)、3个LSTM层和1个Dense层,即1+4+1网络结构,如图5所示。将模型超参数设定为模型迭代次数(iter)、学习率(lr)、第1个LSTM层结点数(Unit1)、第2个LSTM层结点数(Unit2)、第3个LSTM层结点数(Unit3)、Dense层结点数(Unit4),基于训练集采用试验法设置相应的取值范围,并采用交叉验证网格搜索的方式进行超参数寻优,以得到最优的超参数取值,结果如表3所示。
图表编号 | XD00201917500 严禁用于非法目的 |
---|---|
绘制时间 | 2021.02.20 |
作者 | 李增良 |
绘制单位 | 中铁二十局集团有限公司 |
更多格式 | 高清、无水印(增值服务) |