конвертировать conv1D в Conv2D с помощью GRU - PullRequest
Новая
       17

конвертировать conv1D в Conv2D с помощью GRU

0 голосов
/ 14 июня 2019

Я учусь конвертировать мою архитектуру CNN 1D с GRU в CNN2 с GRU.

Вот код в Python 3 с керами Tensorflow. 

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)

Спасибо за вашу помощь

...