def biLSTM_encoder(input, units, dropout, recurrent_dropout, num_layers):
"""Question and context encoder. Just Bi-LSTM from keras."""
encoder = input
for i in range(num_layers):
encoder = Bidirectional(LSTM(units=units,
activation='tanh',
recurrent_activation='hard_sigmoid',
use_bias=True,
kernel_initializer='glorot_uniform',
recurrent_initializer='orthogonal',
bias_initializer='zeros',
unit_forget_bias=True,
kernel_regularizer=None,
recurrent_regularizer=None,
bias_regularizer=None,
activity_regularizer=None,
kernel_constraint=None,
recurrent_constraint=None,
bias_constraint=None,
return_sequences=True,
dropout=dropout,
recurrent_dropout = recurrent_dropout,
unroll=False)) (encoder)
return encoder
评论列表
文章目录