Keras 分子活动模型

Question

我正在尝试 Merku 分子活动挑战，并创建了训练和测试数据集。

数据的形状如下：

x_train.shape=(1452, 4306)
y_train.shape=(1452, 1) 
x_test.shape=(363, 4306)
y_test.shape=(363, 1)

我使用Dense层来定义模型如下：

model = Sequential()
model.add(Dense(100, activation="relu",  input_shape=(4306,)))
model.add(Dense(50, activation="relu"))
model.add(Dense(25, activation="relu"))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(1))

# Compile the model
model.compile(
    loss='categorical_crossentropy',
    optimizer="adam",
)
model.summary()
# Train the model
model.fit(
    x_train,
    y_train,
    batch_size=300,
    epochs=900,
    validation_data=(x_test, y_test),
    shuffle=True
)

在尝试上述代码时，出现以下错误：

ValueError: Input 0 is incompatible with layer flatten_23: expected min_ndim=3, found ndim=2

我该如何解决这个错误？

Answer 1

只需移除 flatten 层：

model = Sequential()
model.add(Dense(100, activation="relu",  input_shape=(4306,)))
model.add(Dense(50, activation="relu"))
model.add(Dense(25, activation="relu"))
model.add(Dropout(0.25))
model.add(Dense(1))

发送到顺序层的数据本质上是一维的（忽略批处理列），因此没有什么可以展平的。进入flatten层的数据已经是1D了。

编辑——回归：

分类交叉熵不是回归的合适成本函数，您需要使用均方误差，它通常用于所有回归任务：

model.compile(
    loss='mse',
    optimizer="adam",
)