如何在 Streamlit 中正确加载训练好的 LSTM 模型

2026-05-21 Python

本文详解因 Keras/TensorFlow 版本不兼容导致的 load_model() 报错问题，提供跨环境（如 Colab → 本地 Streamlit）模型加载的完整解决方案，包括保存/加载最佳实践、版本验证方法及健壮性封装建议。

本文详解因 keras/tensorflow 版本不兼容导致的 `load_model()` 报错问题，提供跨环境（如 colab → 本地 streamlit）模型加载的完整解决方案，包括保存/加载最佳实践、版本验证方法及健壮性封装建议。

在将 Colab 训练的 LSTM 模型部署到本地 Streamlit 应用时，常见的 ValueError: Could not interpret initializer identifier 错误（如涉及 'Orthogonal' 初始化器解析失败），根本原因并非代码逻辑错误，而是 Keras/TensorFlow 版本不一致引发的序列化兼容性问题。Colab 默认使用较旧的 tensorflow==2.15 或 keras==2.13 等版本，而本地 Python 环境可能已升级至 keras>=2.15（独立 Keras 包）或 tensorflow>=2.16，其模型保存格式（.keras）的元数据结构与初始化器注册机制发生了变化。

✅ 正确做法：统一环境 + 显式指定加载方式

1. 优先使用 tf.keras（推荐）

确保训练与加载均使用 同一 TensorFlow 生态，避免混用独立 keras 包：

# ✅ 训练时（Colab 或本地）—— 使用 tf.keras
import tensorflow as tf
from tensorflow.keras.layers import Dense, Dropout, LSTM
from tensorflow.keras.models import Sequential

model = Sequential()
model.add(LSTM(units=50, activation='relu', return_sequences=True, 
               input_shape=(x_train.shape[1], 1)))
model.add(Dropout(0.2))
# ... 其他层（注意：return_sequences=True 应为布尔值 True，非字符串）
model.add(Dense(units=1))
model.compile(optimizer='adam', loss='mse')
model.fit(x_train, y_train, epochs=100)

# ✅ 保存为 TensorFlow 原生格式（兼容性更强）
model.save('lstm_model_tf')  # 不带扩展名 → 保存为 SavedModel 目录
# 或显式指定格式
# model.save('lstm_model.keras', save_format='keras')  # 仅当版本严格一致时安全

2. Streamlit 加载代码（增强鲁棒性）

import streamlit as st
import tensorflow as tf

@st.cache_resource
def load_trained_model():
    try:
        # 优先尝试加载 SavedModel 格式（最兼容）
        model = tf.keras.models.load_model('lstm_model_tf')
        st.success("✅ LSTM 模型加载成功（SavedModel 格式）")
        return model
    except Exception as e1:
        try:
            # 回退：尝试 .keras 格式（需版本匹配）
            model = tf.keras.models.load_model('LSTM_model.keras')
            st.warning("⚠️ 使用 .keras 格式加载 —— 请确保训练/运行环境 Keras 版本一致")
            return model
        except Exception as e2:
            st.error(f"❌ 模型加载失败：\n- SavedModel 错误：{e1}\n- .keras 错误：{e2}")
            st.stop()

# 在 Streamlit 主程序中调用
model = load_trained_model()

3. 关键注意事项

修正代码细节：原训练代码中 return_sequences='True' 是错误写法，应为 return_sequences=True（布尔值），否则可能导致模型结构异常；

验证版本一致性：在 Colab 和本地分别运行：

import tensorflow as tf
print("TensorFlow 版本:", tf.__version__)
print("Keras 版本:", tf.keras.__version__)

确保二者完全相同（例如均为 2.15.1）；

避免混用包：卸载独立 keras 包（pip uninstall keras），仅依赖 tensorflow 自带的 tf.keras；
路径检查：Streamlit 运行时工作目录即 main.py 所在目录，确认 LSTM_model.keras 或 lstm_model_tf/ 目录存在且可读；
缓存优化：使用 @st.cache_resource 装饰器避免每次交互重复加载模型，提升响应速度。