机车能耗预测模型训练方法、装置

本申请涉及机车能耗预测模型训练方法、装置。所述方法包括：采用多个机车能耗影响指标与样本机车能耗数据进行相关性分析，基于相关性分析结果构建采集样本数据序列；通过对采集样本数据序列进行数据预处理，得到模型训练数据集和模型测试数据集；采用鲸鱼优化算法优化用于机车能耗预测的初始模型，获得优化后模型，根据模型训练数据集训练优化后模型，得到机车能耗预测模型；初始模型为结合注意力机制的双向长短期记忆网络构建得到；基于模型测试数据集和机车能耗预测模型，获得输出预测结果，对输出预测结果进行反归一化处理，得到测试预测结果，以验证机车能耗预测模型的有效性。采用本方法能够有效提升机车能耗预测模型的准确性与有效性。