tensorflow - 如何修复 Keras 层输入已在 google colab 中折旧的警告

Question

大家好，我目前正在谷歌 colab 学习机器学习，现在我遇到这个错误警告：tensorflow：请添加keras.layers.InputLayer而不是keras.Input顺序模型。keras.Input旨在供功能模型使用。在谷歌 colab 中。

这是我的导入类：

# Import classes
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Input, Dense, Dropout

这是我的模型定义：

# Model definition
model = Sequential()
model.add(Input(shape=(8,)))
model.add(Dense(16,activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(8,activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1,activation='sigmoid'))

标签： tensorflowkeras

Answer 1

通常建议通过Input, （创建一个InputLayer）使用功能层 API，而不直接使用InputLayer.

与 Keras Sequential 模型一起使用InputLayer时，可以通过将 input_shape 参数移动到InputLayer. 有关更多详细信息，您可以参考此。