基于自注意力机制的能耗预测方法、装置、设备及介质

本申请公开了基于自注意力机制的能耗预测方法、装置、设备及介质，涉及能源消耗预测技术领域，包括：构建目标训练集，并设置与历史月度电量数据对应的各月份分别对应的目标权重值；构造目标损失函数，目标损失函数包括季节性变化幅度惩罚函数、时间步变化一致性损失函数和加权损失函数，并将目标损失函数添加至预设初始长短期记忆网络模型，以获取相应的目标长短期记忆网络模型；根据目标训练集和自注意力机制对目标长短期记忆网络模型进行训练，以获取相应的训练后模型，并利用训练后模型和自注意力机制进行能耗预测。通过设置不同月份分别对应的权重值，并利用长短期记忆网络进行能耗预测，保证了预测结果的准确性。