模型参数

参数名称

参数取值

epoch

迭代次数

100

Batch_size

每批训练样本容量

128

validation_spli

训练验证集

0.2

Lr

学习率

0.001