|
|
|
@ -1,13 +1,17 @@
|
|
|
|
import sys
|
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
|
|
import paddle.v2 as paddle
|
|
|
|
import paddle.v2 as paddle
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def seqToseq_net(source_dict_dim, target_dict_dim):
|
|
|
|
def seqToseq_net(source_dict_dim, target_dict_dim, is_generating=False):
|
|
|
|
### Network Architecture
|
|
|
|
### Network Architecture
|
|
|
|
word_vector_dim = 512 # dimension of word vector
|
|
|
|
word_vector_dim = 512 # dimension of word vector
|
|
|
|
decoder_size = 512 # dimension of hidden unit in GRU Decoder network
|
|
|
|
decoder_size = 512 # dimension of hidden unit in GRU Decoder network
|
|
|
|
encoder_size = 512 # dimension of hidden unit in GRU Encoder network
|
|
|
|
encoder_size = 512 # dimension of hidden unit in GRU Encoder network
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
beam_size = 3
|
|
|
|
|
|
|
|
max_length = 250
|
|
|
|
|
|
|
|
|
|
|
|
#### Encoder
|
|
|
|
#### Encoder
|
|
|
|
src_word_id = paddle.layer.data(
|
|
|
|
src_word_id = paddle.layer.data(
|
|
|
|
name='source_language_word',
|
|
|
|
name='source_language_word',
|
|
|
|
@ -67,6 +71,7 @@ def seqToseq_net(source_dict_dim, target_dict_dim):
|
|
|
|
group_input2 = paddle.layer.StaticInputV2(input=encoded_proj, is_seq=True)
|
|
|
|
group_input2 = paddle.layer.StaticInputV2(input=encoded_proj, is_seq=True)
|
|
|
|
group_inputs = [group_input1, group_input2]
|
|
|
|
group_inputs = [group_input1, group_input2]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if not is_generating:
|
|
|
|
trg_embedding = paddle.layer.embedding(
|
|
|
|
trg_embedding = paddle.layer.embedding(
|
|
|
|
input=paddle.layer.data(
|
|
|
|
input=paddle.layer.data(
|
|
|
|
name='target_language_word',
|
|
|
|
name='target_language_word',
|
|
|
|
@ -91,37 +96,58 @@ def seqToseq_net(source_dict_dim, target_dict_dim):
|
|
|
|
cost = paddle.layer.classification_cost(input=decoder, label=lbl)
|
|
|
|
cost = paddle.layer.classification_cost(input=decoder, label=lbl)
|
|
|
|
|
|
|
|
|
|
|
|
return cost
|
|
|
|
return cost
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
# In generation, the decoder predicts a next target word based on
|
|
|
|
|
|
|
|
# the encoded source sequence and the last generated target word.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# The encoded source sequence (encoder's output) must be specified by
|
|
|
|
|
|
|
|
# StaticInput, which is a read-only memory.
|
|
|
|
|
|
|
|
# Embedding of the last generated word is automatically gotten by
|
|
|
|
|
|
|
|
# GeneratedInputs, which is initialized by a start mark, such as <s>,
|
|
|
|
|
|
|
|
# and must be included in generation.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
trg_embedding = paddle.layer.GeneratedInputV2(
|
|
|
|
|
|
|
|
size=target_dict_dim,
|
|
|
|
|
|
|
|
embedding_name='_target_language_embedding',
|
|
|
|
|
|
|
|
embedding_size=word_vector_dim)
|
|
|
|
|
|
|
|
group_inputs.append(trg_embedding)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
beam_gen = paddle.layer.beam_search(
|
|
|
|
|
|
|
|
name=decoder_group_name,
|
|
|
|
|
|
|
|
step=gru_decoder_with_attention,
|
|
|
|
|
|
|
|
input=group_inputs,
|
|
|
|
|
|
|
|
bos_id=0,
|
|
|
|
|
|
|
|
eos_id=1,
|
|
|
|
|
|
|
|
beam_size=beam_size,
|
|
|
|
|
|
|
|
max_length=max_length)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return beam_gen
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
def main():
|
|
|
|
paddle.init(use_gpu=False, trainer_count=1)
|
|
|
|
paddle.init(use_gpu=False, trainer_count=1)
|
|
|
|
|
|
|
|
is_generating = False
|
|
|
|
|
|
|
|
|
|
|
|
# source and target dict dim.
|
|
|
|
# source and target dict dim.
|
|
|
|
dict_size = 30000
|
|
|
|
dict_size = 30000
|
|
|
|
source_dict_dim = target_dict_dim = dict_size
|
|
|
|
source_dict_dim = target_dict_dim = dict_size
|
|
|
|
|
|
|
|
|
|
|
|
# define network topology
|
|
|
|
# train the network
|
|
|
|
|
|
|
|
if not is_generating:
|
|
|
|
cost = seqToseq_net(source_dict_dim, target_dict_dim)
|
|
|
|
cost = seqToseq_net(source_dict_dim, target_dict_dim)
|
|
|
|
parameters = paddle.parameters.create(cost)
|
|
|
|
parameters = paddle.parameters.create(cost)
|
|
|
|
|
|
|
|
|
|
|
|
# define optimize method and trainer
|
|
|
|
# define optimize method and trainer
|
|
|
|
optimizer = paddle.optimizer.Adam(
|
|
|
|
optimizer = paddle.optimizer.Adam(
|
|
|
|
learning_rate=5e-5,
|
|
|
|
learning_rate=5e-5,
|
|
|
|
regularization=paddle.optimizer.L2Regularization(rate=1e-3))
|
|
|
|
regularization=paddle.optimizer.L2Regularization(rate=8e-4))
|
|
|
|
trainer = paddle.trainer.SGD(cost=cost,
|
|
|
|
trainer = paddle.trainer.SGD(cost=cost,
|
|
|
|
parameters=parameters,
|
|
|
|
parameters=parameters,
|
|
|
|
update_equation=optimizer)
|
|
|
|
update_equation=optimizer)
|
|
|
|
|
|
|
|
|
|
|
|
# define data reader
|
|
|
|
# define data reader
|
|
|
|
feeding = {
|
|
|
|
|
|
|
|
'source_language_word': 0,
|
|
|
|
|
|
|
|
'target_language_word': 1,
|
|
|
|
|
|
|
|
'target_language_next_word': 2
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
wmt14_reader = paddle.batch(
|
|
|
|
wmt14_reader = paddle.batch(
|
|
|
|
paddle.reader.shuffle(
|
|
|
|
paddle.reader.shuffle(
|
|
|
|
paddle.dataset.wmt14.train(dict_size=dict_size), buf_size=8192),
|
|
|
|
paddle.dataset.wmt14.train(dict_size), buf_size=8192),
|
|
|
|
batch_size=5)
|
|
|
|
batch_size=5)
|
|
|
|
|
|
|
|
|
|
|
|
# define event_handler callback
|
|
|
|
# define event_handler callback
|
|
|
|
@ -129,17 +155,59 @@ def main():
|
|
|
|
if isinstance(event, paddle.event.EndIteration):
|
|
|
|
if isinstance(event, paddle.event.EndIteration):
|
|
|
|
if event.batch_id % 10 == 0:
|
|
|
|
if event.batch_id % 10 == 0:
|
|
|
|
print "\nPass %d, Batch %d, Cost %f, %s" % (
|
|
|
|
print "\nPass %d, Batch %d, Cost %f, %s" % (
|
|
|
|
event.pass_id, event.batch_id, event.cost, event.metrics)
|
|
|
|
event.pass_id, event.batch_id, event.cost,
|
|
|
|
|
|
|
|
event.metrics)
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
sys.stdout.write('.')
|
|
|
|
sys.stdout.write('.')
|
|
|
|
sys.stdout.flush()
|
|
|
|
sys.stdout.flush()
|
|
|
|
|
|
|
|
|
|
|
|
# start to train
|
|
|
|
# start to train
|
|
|
|
trainer.train(
|
|
|
|
trainer.train(
|
|
|
|
reader=wmt14_reader,
|
|
|
|
reader=wmt14_reader, event_handler=event_handler, num_passes=2)
|
|
|
|
event_handler=event_handler,
|
|
|
|
|
|
|
|
num_passes=10000,
|
|
|
|
# generate a english sequence to french
|
|
|
|
feeding=feeding)
|
|
|
|
else:
|
|
|
|
|
|
|
|
# use the first 3 samples for generation
|
|
|
|
|
|
|
|
gen_creator = paddle.dataset.wmt14.gen(dict_size)
|
|
|
|
|
|
|
|
gen_data = []
|
|
|
|
|
|
|
|
gen_num = 3
|
|
|
|
|
|
|
|
for item in gen_creator():
|
|
|
|
|
|
|
|
gen_data.append((item[0], ))
|
|
|
|
|
|
|
|
if len(gen_data) == gen_num:
|
|
|
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
beam_gen = seqToseq_net(source_dict_dim, target_dict_dim, is_generating)
|
|
|
|
|
|
|
|
# get the pretrained model, whose bleu = 26.92
|
|
|
|
|
|
|
|
parameters = paddle.dataset.wmt14.model()
|
|
|
|
|
|
|
|
# prob is the prediction probabilities, and id is the prediction word.
|
|
|
|
|
|
|
|
beam_result = paddle.infer(
|
|
|
|
|
|
|
|
output_layer=beam_gen,
|
|
|
|
|
|
|
|
parameters=parameters,
|
|
|
|
|
|
|
|
input=gen_data,
|
|
|
|
|
|
|
|
field=['prob', 'id'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# get the dictionary
|
|
|
|
|
|
|
|
src_dict, trg_dict = paddle.dataset.wmt14.get_dict(dict_size)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# the delimited element of generated sequences is -1,
|
|
|
|
|
|
|
|
# the first element of each generated sequence is the sequence length
|
|
|
|
|
|
|
|
seq_list = []
|
|
|
|
|
|
|
|
seq = []
|
|
|
|
|
|
|
|
for w in beam_result[1]:
|
|
|
|
|
|
|
|
if w != -1:
|
|
|
|
|
|
|
|
seq.append(w)
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
seq_list.append(' '.join([trg_dict.get(w) for w in seq[1:]]))
|
|
|
|
|
|
|
|
seq = []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
prob = beam_result[0]
|
|
|
|
|
|
|
|
beam_size = 3
|
|
|
|
|
|
|
|
for i in xrange(gen_num):
|
|
|
|
|
|
|
|
print "\n*******************************************************\n"
|
|
|
|
|
|
|
|
print "src:", ' '.join(
|
|
|
|
|
|
|
|
[src_dict.get(w) for w in gen_data[i][0]]), "\n"
|
|
|
|
|
|
|
|
for j in xrange(beam_size):
|
|
|
|
|
|
|
|
print "prob = %f:" % (prob[i][j]), seq_list[i * beam_size + j]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
typhoonzero
8 years ago