def __init__(self, n_layers, n_source_vocab, n_target_vocab, n_source_char, n_units, n_sentences):
super(Seq2seq, self).__init__(
embed_xw=L.EmbedID(n_source_vocab, n_units),
embed_xc=L.EmbedID(n_source_char, n_units),
embed_y=L.EmbedID(n_target_vocab, n_units * 2),
encoder_fw=L.NStepGRU(n_layers, n_units, n_units, 0.1),
encoder_bw=L.NStepGRU(n_layers, n_units, n_units, 0.1),
encoder_fc=L.NStepGRU(n_layers, n_units, n_units, 0.1),
encoder_bc=L.NStepGRU(n_layers, n_units, n_units, 0.1),
decoder=My.NStepGRU(n_layers, n_units * 2, n_units * 2, 0.1),
W=L.Linear(n_units * 2, n_target_vocab),
)
self.n_layers = n_layers
self.n_units = n_units
self.n_params = 6
self.n_sentences = n_sentences
self.n_process = 0
self.n_sen = len(str(n_sentences))
评论列表
文章目录