使用GRU将conv1D转换为Conv2D

时间:2019-06-14 15:43:56

标签: python-3.x machine-learning keras keras-layer

我正在学习将具有GRU的CNN 1D结构转换为具有GRU的CNN2。

这是带有Tensorflow keras的python 3中的代码。

model = Sequential()
model.add(layers.Conv1D(32, 5, activation='relu',
                        input_shape=(None, float_data.shape[-1])))
model.add(layers.MaxPooling1D(3))
model.add(layers.Conv1D(32, 5, activation='relu'))
model.add(layers.GRU(32, dropout=0.1, recurrent_dropout=0.5))
model.add(layers.Dense(1))

model.summary()

model.compile(optimizer=RMSprop(), loss='mae')
history = model.fit_generator(train_gen,
                              steps_per_epoch=500,
                              epochs=20,
                              validation_data=val_gen,
                              validation_steps=val_steps)

感谢您的帮助

0 个答案:

没有答案