如何在tf.keras的输出层中指定单位？

Question

在我的数据集中，单个输入和输出的形状为15 x 61。我指定的是input_shape =（15,61）。这意味着在15 x 61到15 x 61之间映射，这不是分类问题。输出层中的单位规格应该是什么？

model = tf.keras.models.Sequential([tf.keras.layers.Dense(units=128,input_shape =(15,61),
tf.keras.layers.Dense(units=128, activation=tf.nn.relu),
tf.keras.layers.Dense(units=, activation=tf.nn.softmax)])

Answer 1

您在Keras turorial on Dense中拥有所有信息。如果要进行分类，则输出形状必须是类数。 如果希望将15 * 61作为输入，将15 * 61作为输出，则将所有内容展平：15 x 61 = 915，因此，请使用915尺寸的图层并输出915尺寸的图层，然后对其进行重塑以使其具有初始形状。形状。