1. 概述1.1 项目背景EMS（Energy Management System）能源管理系统的 AI 预测服务，基于 PyTorch 构建深度学习模型，实现能源消耗的时序预测。系统采用 LSTM（长短期记忆网络）算法，能够捕捉时间序列数据中的长期依赖关系，为企业提供高精度的能源消耗预测能力。1.2 核心功能模型训练：基于历史能耗数据训练 LSTM 模型时序预测：支持 24 小时多步滚动预测模型管理：支持模型版本管理和加载性能评估：提供 RMSE、MAE、MAPE 等评估指标1.3 技术栈【核心框架】FastAPI 0.104.1 + PyTorch 2.1.2 + NumPy 1.26.2 + pandas 2.1.4【部署方式】Uvicorn ASGI 服务器，独立微服务架构2. LSTM 算法原理2.1 LSTM 网络结构LSTM（Long Short-Term Memory）是一种特殊的循环神经网络（RNN），通过门控机制解决了传统 RNN 的梯度消失问题，能够有效捕捉时间序列中的长期依赖关系。2.2 门控机制LSTM 单元包含三个门控：遗忘门（Forget Gate）：决定丢弃哪些信息f_t = σ(W_f · [h_{t-1}, x_t] + b_f)输入门（Input Gate）：决定存储哪些新信息i_t = σ(W_i · [h_{t-1}, x_t] + b_i)C̃_t = tanh(W_C · [h_{t-1}, x_t] + b_C)输出门（Output Gate）：决定输出哪些信息o_t = σ(W_o · [h_{t-1}, x_t] + b_o)h_t = o_t · tanh(C_t)2.3 单元状态更新C_t = f_t · C_{t-1} + i_t · C̃_t其中：C_t 为当前单元状态，C_{t-1} 为前一时刻单元状态，f_t 为遗忘门输出，i_t 为输入门输出，C̃_t 为候选状态。2.4 为什么选择 LSTM能够捕捉时间序列中的长期依赖关系通过门控机制解决梯度消失问题适合处理连续的时序数据（如每小时能耗数据）对噪声有较好的鲁棒性3. 模型架构设计3.1 模型结构本系统采用双层 LSTM 架构，结构如下：输入层 → LSTM 层 1(64) → LSTM 层 2(64) → ReLU 激活 → 全连接层 → 输出层3.2 模型参数【参数配置】input_size = 1# 单特征输入（能耗值）hidden_size = 64# 隐藏层维度num_layers = 2# LSTM层数output_size = 1# 单值输出（预测值）dropout = 0.2# Dropout正则化率3.3 模型类定义【关键代码】lstm_model.pyclass EnergyLSTMModel(nn.Module):def __init__(self, input_size=1, hidden_size=64, num_layers=2,output_size=1, dropout=0.2):super(EnergyLSTMModel, self).__init__()self.hidden_size = hidden_sizeself.num_layers = num_layersself.lstm = nn.LSTM(input_size, hidden_size, num_layers,batch_first=True, dropout=dropout)self.fc = nn.Linear(hidden_size, output_size)self.relu = nn.ReLU()def forward(self, x):h0 = torch.zeros(self.num_layers, x.size(0),self.hidden_size).to(x.device)c0 = torch.zeros(self.num_layers, x.size(0),self.hidden_size).to(x.device)out, _ = self.lstm(x, (h0, c0))out = self.fc(self.relu(out[:, -1, :]))return out3.4 前向传播流程输入数据形状：(batch_size, seq_length, input_size)初始化h0和c0：(num_layers, batch_size, hidden_size)LSTM层输出：(batch_size, seq_length, hidden_size)取最后时间步：(batch_size, hidden_size)ReLU激活全连接层输出：(batch_size, output_size)4. 数据预处理设计4.1 数据格式输入数据为时序能耗记录列表，每条记录包含：{ "timestamp": "2024-01-01T00:00:00", "calculatedValue": 125.5 }4.2 序列创建算法【滑动窗口法】使用固定长度的滑动窗口从时序数据中创建输入输出对。窗口长度（seq_length）= 24，表示使用过去24小时数据预测下一小时。4.3 序列创建代码