创建LSTM模型
model = Sequential()
model.add(LSTM(48, activation='relu', input_shape=(features_train.shape[1], features_train.shape[2])))
model.add(Dense(1)) # 输出层只有一个神经元
编译模型
model.compile(optimizer='adam', loss='mse')
设置回调
early_stopping = EarlyStopping(monitor='val_loss', patience=20, restore_best_weights=True)
model.load_weights('') # 加载最佳权重
train_score = model.evaluate(features_train, target_train, verbose=0)
test_score = model.evaluate(features_val, target_val, verbose=0)