这是您上面提供的代码的模型摘要:
Model: "sequential_1"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
conv1d (Conv1D) (None, 3023, 16) 64
_________________________________________________________________
conv1d_1 (Conv1D) (None, 3021, 16) 784
_________________________________________________________________
max_pooling1d (MaxPooling1D) (None, 1510, 16) 0
_________________________________________________________________
dropout (Dropout) (None, 1510, 16) 0
_________________________________________________________________
conv1d_2 (Conv1D) (None, 1508, 32) 1568
_________________________________________________________________
conv1d_3 (Conv1D) (None, 1506, 32) 3104
_________________________________________________________________
max_pooling1d_1 (MaxPooling1 (None, 753, 32) 0
_________________________________________________________________
dropout_1 (Dropout) (None, 753, 32) 0
_________________________________________________________________
dense (Dense) (None, 753, 500) 16500
_________________________________________________________________
dense_1 (Dense) (None, 753, 300) 150300
_________________________________________________________________
dense_2 (Dense) (None, 753, 4) 1204
=================================================================
Total params: 173,524
Trainable params: 173,524
Non-trainable params: 0
输出层的维度为(批次、序列长度、4 类)。您可能打算在第二个 max_pooling 层之后展平矩阵。
如果我这样做,我会得到一个参数较少的模型,并将输出 4 个类之一...
def cnn_1d(num_classes):
model = models.Sequential()
model.add(Conv1D(16, kernel_size=3, strides=1, activation="relu", input_shape=(3025, 1)))
model.add(Conv1D(16, kernel_size=3, strides=1, activation="relu"))
model.add(MaxPooling1D(pool_size=2))
model.add(Dropout(0.2))
model.add(Conv1D(32, kernel_size=3, strides=1, activation="relu"))
model.add(Conv1D(32, kernel_size=3, strides=1, activation="relu"))
model.add(MaxPooling1D(pool_size=2))
model.add(Flatten())
model.add(Dropout(0.2))
model.add(Dense(500, activation="relu"))
model.add(Dense(300, activation="relu"))
model.add(Dense(num_classes, activation="softmax"))
model.compile(
optimizer=keras.optimizers.Adam(1e-3),
loss="categorical_crossentropy",
metrics=["accuracy"],
)
model.summary()
return model
cnn_1d(4)
Model: "sequential_2"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
conv1d_4 (Conv1D) (None, 3023, 16) 64
_________________________________________________________________
conv1d_5 (Conv1D) (None, 3021, 16) 784
_________________________________________________________________
max_pooling1d_2 (MaxPooling1 (None, 1510, 16) 0
_________________________________________________________________
dropout_2 (Dropout) (None, 1510, 16) 0
_________________________________________________________________
conv1d_6 (Conv1D) (None, 1508, 32) 1568
_________________________________________________________________
conv1d_7 (Conv1D) (None, 1506, 32) 3104
_________________________________________________________________
max_pooling1d_3 (MaxPooling1 (None, 753, 32) 0
_________________________________________________________________
flatten (Flatten) (None, 24096) 0
_________________________________________________________________
dropout_3 (Dropout) (None, 24096) 0
_________________________________________________________________
dense_3 (Dense) (None, 500) 12048500
_________________________________________________________________
dense_4 (Dense) (None, 300) 150300
_________________________________________________________________
dense_5 (Dense) (None, 4) 1204
=================================================================
Total params: 12,205,524
Trainable params: 12,205,524
Non-trainable params: 0
作为奖励,此模型的可训练参数显着减少。