如何在 Sequential keras 模型中定义输入形状

Question

请帮助在 keras 模型中定义适当的 Dense 输入形状。也许我必须先重塑我的数据。我的数据集的维度如下所示：

Data shapes are X_train: (2858, 2037) y_train: (2858, 1) X_test: (715, 2037) y_test: (715, 1) Number of features (input shape) is 2037

我想像那样定义 Sequential keras 模型

``

    batch_size = 128
    num_classes = 2
    epochs = 20

    model = Sequential()
    model.add(Dense(512, activation='relu', input_shape=(X_input_shape,)))
    model.add(Dropout(0.2))
    model.add(Dense(512, activation='relu'))

    model.summary()

    model.compile(loss='binary_crossentropy',
                optimizer=RMSprop(),
                from_logits=True,
                metrics=['accuracy'])

``

模型摘要：

``

    Layer (type)                 Output Shape              Param #   
    =================================================================
    dense_20 (Dense)             (None, 512)               1043456   
    _________________________________________________________________
    dropout_12 (Dropout)         (None, 512)               0         
    _________________________________________________________________
    dense_21 (Dense)             (None, 512)               262656    
    =================================================================
    Total params: 1,306,112
    Trainable params: 1,306,112
    Non-trainable params: 0

``

当我尝试适应它时...

``

    history = model.fit(X_train, y_train,
                        batch_size=batch_size,
                        epochs=epochs,
                        verbose=1,
                        validation_data=(X_test, y_test))

`` 我收到一个错误：

``

  ValueError: Error when checking target: expected dense_21 to have shape (512,) but got array with shape (1,)

``

Answer 1

修改

model.add(Dense(512, activation='relu'))

至

model.add(Dense(1, activation='relu'))

大小为 1 的输出形状，与 y_train.shape[1] 相同。

如何在 Sequential keras 模型中定义输入形状

How to define inpute shapes in Sequential keras model

python

keras-layer