TensorFlow项目导入Keras回调组件遇Pylance导入解析错误

问题描述 #

在TensorFlow项目中尝试导入EarlyStopping、ModelCheckpoint等Keras回调组件时，使用以下导入语句：

from tensorflow.keras.callbacks import EarlyStopping ModelCheckpoint

或

from keras.callbacks import EarlyStopping ModelCheckpoint

始终收到Pylance报错：Import "keras.callbacks" could not be resolved PylancereportMissingImport，即使已成功安装TensorFlow和Keras，错误仍未消失。

解决方案 #

Keras 3.x版本调整了库结构，需从keras.src子目录导入相关组件。修改后的有效导入语句为：

from keras.src.callbacks import EarlyStopping
from keras.src.callbacks import ModelCheckpoint

也可以合并为一行导入：

from keras.src.callbacks import EarlyStopping, ModelCheckpoint

修改后导入正常，Pylance不再提示缺失导入错误。

环境说明 #

• TensorFlow版本：2.17.0（通过pip install tensorflow安装）
• Keras版本：3.5.0
• IDE：Visual Studio Code
• 问题特定于Pylance语言服务

内容的提问来源于stack exchange，提问作者Mohit Nikhade