在keras函数API中将3D输入转换为2D输出

Question

我有三个Numpy数组，每个数组的形状都是(6000,3256(。我想使用这三个作为Keras Functional API用于密集神经网络的输入。我的输出应该是正常的(6000,2(。我对如何指定尺寸感到困惑。

Answer 1

密集层或全连接层采用一维输入，因此需要压平输入数组。

X = np.reshape(6000, 3*256)
ip = Input(shape=(3*256))
dense1 = Dense(128, activation = 'relu')(ip)
dense2 = Dense(2, activation = 'sigmoid')(dense1)

您需要根据需要选择中间层、损失函数和激活。