Я передаю временную последовательность или вектор? - PullRequest
0 голосов
/ 29 марта 2019

Я пытаюсь передать список из трех элементов для встраивания, а затем в LSTM как временную последовательность, но я не знаю, передаю ли я только значение, которое является вектором 3. Это это код:

lr = 1e-3
emb_size = 100
look_back = 3

# "Expert" (regular dqn) model architecture

inp = Input(shape=(look_back,))
emb = Embedding(input_dim=env.action_space.n+1, output_dim = emb_size)(inp) 
rnn = Bidirectional(LSTM(5))(emb)
out = Dense(env.action_space.n, activation='softmax')(rnn)
expert_model = Model(inputs = inp, outputs = out)
expert_model.compile(loss='categorical_crossentropy', optimizer= Adam(lr))

print(expert_model.summary())

И это резюме:

_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
input_1 (InputLayer)         (None, 3)                 0         
_________________________________________________________________
embedding_1 (Embedding)      (None, 3, 100)            1100      
_________________________________________________________________
bidirectional_1 (Bidirection (None, 10)                4240      
_________________________________________________________________
dense_1 (Dense)              (None, 10)                110       
=================================================================
Total params: 5,450
Trainable params: 5,450
Non-trainable params: 0
_________________________________________________________________
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...