def __init__(self, n_enc_vocab, n_dec_vocab, n_hidden, bidirectional,
linearTransform):
super(Seq2SeqGRU_Attention, self).__init__()
self.n_hidden = n_hidden
self.num_directions = 2 if bidirectional is True else 1
self.encoder = nb.GRU(input_size=n_enc_vocab,
hidden_size=n_hidden,
bidirectional=bidirectional)
self.decoder = nb.GRU(input_size=n_dec_vocab,
hidden_size=n_hidden,
bidirectional=bidirectional)
self.attention = nb.AttentionTwo(n_dec_vocab,
n_hidden,
bidirectional=bidirectional,
linearTransform=linearTransform)
def __init__(self,
emb_dim,
n_class,
n_hidden,
bidirectional=False,
linearTransform=True):
super(GRU_Attention, self).__init__()
self.n_hidden = n_hidden
self.num_directions = 2 if bidirectional is True else 1
self.gru = nb.GRU(emb_dim, n_hidden, bidirectional=bidirectional)
self.attention = nb.AttentionOne(n_class,
n_hidden,
bidirectional=bidirectional,
linearTransform=linearTransform)