检查输入时出错：预期density_input具有3维

Question

我有一个数据，我正在尝试使其与Keras模型和我的train_data形状（161,25）匹配

model=keras.Sequential([
keras.layers.Dense(64,activation=tf.nn.relu,input_shape=(train_data.shape)),
keras.layers.Dense(64,activation=tf.nn.relu),
keras.layers.Dense(1)])

现在，当我想按以下方式拟合train_data时

model.fit(train_data,train_labels,epochs=500,validation_split=0.02,verbose=0)

说

Error when checking input: expected dense_input to have 3 dimensions, but got array with shape (161, 25)

有什么建议吗？

Answer 1

我发现了问题，如果有人也有此问题，我在这里回答： 所以我的输入形状为（161,25），这意味着我们有161个数据，每个数据都有25个特征，并且我们希望将每个数据的每个25个特征提供给网络，因此我们将第一层更改为：

keras.layers.Dense(64,activation=tf.nn.relu,input_shape=(train_data.shape)),

收件人：

keras.layers.Dense(64,activation=tf.nn.relu,input_shape=(train_data.shape[1],)),