import numpy as np
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
import matplotlib.pyplot as plt # 如果需要内部可视化，暂时使用Matplotlib，输出将使用Plotly

# 设置随机种子以保证结果可复现
tf.random.set_seed(42)
np.random.seed(42)

1. 生成和准备数据

首先，我们创建时间序列数据，一个带有噪声的正弦波，使其略具挑战性。

# 生成正弦波数据
time = np.arange(0, 100, 0.1)
amplitude = np.sin(time) + np.random.normal(scale=0.15, size=len(time))

# 可视化生成的数据

已生成带有随机噪声的正弦波。我们将使用此时间序列进行预测任务。

在将数据输入神经网络之前，通常需要进行缩放，通常缩放到0到1或-1到1的范围。这有助于优化算法更有效地收敛。我们将使用MinMaxScaler。

from sklearn.preprocessing import MinMaxScaler

scaler = MinMaxScaler(feature_range=(0, 1))
# 为缩放器重塑数据：(样本数, 特征数)
amplitude_scaled = scaler.fit_transform(amplitude.reshape(-1, 1))

现在，我们需要将这个单一时间序列转换为适合监督学习的输入/输出对。我们将使用滑动窗口方法：将过去值序列（window_size）作为输入（ $X$ ），下一个单一值作为输出（ $y$ ）。

def create_sequences(data, window_size):
    X, y = [], []
    for i in range(len(data) - window_size):
        X.append(data[i:(i + window_size), 0])
        y.append(data[i + window_size, 0])
    return np.array(X), np.array(y)

window_size = 50 # 用于作为输入特征的过去时间步长数量
X, y = create_sequences(amplitude_scaled, window_size)

# 将数据拆分为训练集和测试集（例如，80%训练，20%测试）
train_size = int(len(X) * 0.8)
X_train, X_test = X[:train_size], X[train_size:]
y_train, y_test = y[:train_size], y[train_size:]

print(f"原始数据长度: {len(amplitude)}")
print(f"缩放后数据长度: {len(amplitude_scaled)}")
print(f"序列数量 (X): {len(X)}")
print(f"训练序列: {len(X_train)}, 测试序列: {len(X_test)}")

LSTM（以及TensorFlow/Keras中的其他RNN层）期望输入数据为特定的3D形状：(batch_size, timesteps, features)。

batch_size: 训练时一次处理的序列数量。
timesteps: 每个输入序列的长度（即我们的window_size）。
features: 每个时间步观察到的特征数量。由于我们是单变量时间序列（只有振幅），此值为1。

# 为LSTM输入重塑X：[样本数, 时间步长, 特征数]
X_train = np.reshape(X_train, (X_train.shape[0], X_train.shape[1], 1))
X_test = np.reshape(X_test, (X_test.shape[0], X_test.shape[1], 1))

print(f"X_train shape: {X_train.shape}") # 应为 (训练样本数, 窗口大小, 1)
print(f"X_test shape: {X_test.shape}")   # 应为 (测试样本数, 窗口大小, 1)
print(f"y_train shape: {y_train.shape}") # 应为 (训练样本数,)
print(f"y_test shape: {y_test.shape}")   # 应为 (测试样本数,)

2. 构建LSTM模型

我们将构建一个简单的LSTM模型。它将包含一个LSTM层，后接一个Dense输出层。

LSTM层： 我们使用layers.LSTM。第一个参数是LSTM层中的单元（神经元）数量。这决定了隐藏状态和单元状态的维度。input_shape=(window_size, 1)告知该层预期接收的输入序列形状（此处无需指定批量大小）。
Dense层： 一个标准的、带有一个神经元的全连接层，因为我们正在预测一个单一输出值（时间序列中的下一个点）。对于此类回归任务，激活函数通常是线性的。

model = keras.Sequential(
    [
        layers.LSTM(units=64, input_shape=(window_size, 1)),
        layers.Dropout(0.2), # 添加Dropout进行正则化
        layers.Dense(units=1) # 输出层：预测下一个单一值
    ]
)

model.summary()

我们需要编译模型，指定优化器和损失函数。Adam是一种常见且有效的优化器。由于这是一个回归问题（预测连续值），均方误差（MSE）是一个合适的损失函数。

\text{均方误差} = \frac{1}{N} \sum_{i=1}^{N} (y_{\text{真实}, i} - y_{\text{预测}, i})^2

model.compile(optimizer='adam', loss='mean_squared_error')

3. 训练模型

现在，我们使用准备好的训练数据来训练模型。我们指定训练轮数（遍历整个数据集的次数）和批量大小。我们还使用一部分训练数据作为验证集，以在训练期间监测模型在未见数据上的表现。

history = model.fit(
    X_train,
    y_train,
    epochs=25,
    batch_size=32,
    validation_split=0.1, # 使用10%的训练数据作为验证
    verbose=1 # 显示进度
)

我们可以可视化训练和验证损失以检查收敛和过拟合情况。

训练和验证损失曲线随训练轮数的变化。理想情况下，两者都应降低并收敛。它们之间的大差距可能表明过拟合。

4. 进行预测和评估

模型训练完成后，让我们评估其在保留测试集上的表现。

# 在测试数据上评估模型
test_loss = model.evaluate(X_test, y_test, verbose=0)
print(f"测试损失 (MSE): {test_loss:.6f}")

# 对测试数据进行预测
predictions_scaled = model.predict(X_test)

# 将预测结果和真实值反向缩放回原始比例
predictions = scaler.inverse_transform(predictions_scaled)
y_test_original = scaler.inverse_transform(y_test.reshape(-1, 1))

最后，让我们可视化模型预测值与测试集中实际值的匹配程度。

实际时间序列值（测试集）与LSTM模型预测值的比较。

讨论

这个实践案例演示了使用简单LSTM模型预测正弦波。我们涵盖了数据生成、缩放、序列创建、模型构建（LSTM + Dense）、训练、评估（MSE）以及预测可视化。

结果显示LSTM能相当好地捕捉到潜在模式。提升表现的潜在后续步骤可能包括：

调整超参数（LSTM单元数量、窗口大小、学习率、批量大小）。
尝试更深的模型（堆叠LSTM层）。
使用GRU而非LSTM以比较表现和效率。
尝试不同的优化器或添加更多正则化。 "* 将此工作流程应用于更复杂的时间序列数据。"

此练习为处理各类时间序列预测问题奠定了基础，使用循环神经网络。

这部分内容有帮助吗？