def __init__(self, in_size, out_size, trainer):
self.out_size_ = out_size
self.pwxh_ = Parameter([4 * out_size, in_size], I.Uniform(-0.1, 0.1))
self.pwhh_ = Parameter([4 * out_size, out_size], I.Uniform(-0.1, 0.1))
self.pbh_ = Parameter([4 * out_size], I.Constant(0))
trainer.add_parameter(self.pwxh_)
trainer.add_parameter(self.pwhh_)
trainer.add_parameter(self.pbh_)
def __init__(self, vocab_size, eos_id):
self.eos_id = eos_id
self.plookup = Parameter([NUM_HIDDEN_UNITS, vocab_size], I.Uniform(-0.1, 0.1))
self.rnn1 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS)
self.rnn2 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS)
self.hy = Affine(NUM_HIDDEN_UNITS, vocab_size)
self.scan_attributes()
def __init__(self, in_size, out_size, trainer):
self.out_size_ = out_size
self.pw_ = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1))
self.pbf_ = Parameter([out_size], I.Constant(0))
self.pbr_ = Parameter([out_size], I.Constant(0))
trainer.add_parameter(self.pw_)
trainer.add_parameter(self.pbf_)
trainer.add_parameter(self.pbr_)
def __init__(self, vocab_size, eos_id, trainer):
self.eos_id_ = eos_id
self.plookup_ = Parameter([NUM_HIDDEN_UNITS, vocab_size],
I.Uniform(-0.1, 0.1))
self.rnn1_ = SRU(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS, trainer)
self.rnn2_ = SRU(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS, trainer)
self.hy_ = Affine(NUM_HIDDEN_UNITS, vocab_size, trainer)
trainer.add_parameter(self.plookup_)
def __init__(self, vocab_size, eos_id):
self.eos_id = eos_id
self.plookup = Parameter([NUM_HIDDEN_UNITS, vocab_size], I.Uniform(-0.1, 0.1))
self.rnn1 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS)
self.rnn2 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS)
self.hy = Affine(NUM_HIDDEN_UNITS, vocab_size)
self.add_all_parameters()
self.add_all_submodels()
def __init__(self, in_size, out_size):
self.out_size = out_size
self.pw = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1))
self.pbf = Parameter([out_size], I.Constant(0))
self.pbr = Parameter([out_size], I.Constant(0))
self.scan_attributes()
def __init__(self, in_size, out_size):
self.out_size = out_size
self.pw = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1))
self.pbf = Parameter([out_size], I.Constant(0))
self.pbr = Parameter([out_size], I.Constant(0))
self.add_all_parameters()