def assert_forced_decoding(self, sent_id):
dy.renew_cg()
outputs = self.model.generate(
batchers.mark_as_batch([self.src_data[sent_id]]),
BeamSearch(),
forced_trg_ids=batchers.mark_as_batch([self.trg_data[sent_id]]))
self.assertItemsEqual(self.trg_data[sent_id].words, outputs[0].words)
def test_single(self):
dy.renew_cg()
train_loss = self.model.calc_nll(src=self.src_data[0],
trg=self.trg_data[0]).value()
dy.renew_cg()
outputs = self.model.generate(
batchers.mark_as_batch([self.src_data[0]]),
BeamSearch(beam_size=1),
forced_trg_ids=batchers.mark_as_batch([self.trg_data[0]]))
self.assertAlmostEqual(-outputs[0].score, train_loss, places=4)
def test_greedy_vs_beam(self):
dy.renew_cg()
outputs = self.model.generate(
batchers.mark_as_batch([self.src_data[0]]),
BeamSearch(beam_size=1))
output_score1 = outputs[0].score
dy.renew_cg()
outputs = self.model.generate(
batchers.mark_as_batch([self.src_data[0]]), GreedySearch())
output_score2 = outputs[0].score
self.assertAlmostEqual(output_score1, output_score2)
def test_single(self):
dy.renew_cg()
outputs = self.model.generate(
batchers.mark_as_batch([self.src_data[0]]), BeamSearch())
# Make sure the output of beam search is the same as the target sentence
# (this is a very overfit model on exactly this data)
self.assertEqual(outputs[0].sent_len(), self.trg_data[0].sent_len())
for i in range(outputs[0].sent_len()):
self.assertEqual(outputs[0][i], self.trg_data[0][i])
# Verify that the loss we get from beam search is the same as the loss
# we get if we call model.calc_nll
dy.renew_cg()
train_loss = self.model.calc_nll(src=self.src_data[0],
trg=outputs[0]).value()
self.assertAlmostEqual(-outputs[0].score, train_loss, places=4)
def run(self):
seed = 13
random.seed(seed)
np.random.seed(seed)
EXP_DIR = os.path.dirname(__file__)
EXP = "annot"
model_file = f"{EXP_DIR}/results/{EXP}.mod"
log_file = f"{EXP_DIR}/results/{EXP}.log"
xnmt.tee.utils.dy.DynetParams().set_mem(
1024) #Doesnt work figure out how to set memory
xnmt.tee.set_out_file(log_file, exp_name=EXP)
ParamManager.init_param_col()
ParamManager.param_col.model_file = model_file
pre_runner = PreprocRunner(
tasks=[
PreprocTokenize(
in_files=
[ #f'{EXP_DIR}/conala-corpus/conala-trainnodev.snippet',
#f'{EXP_DIR}/conala-corpus/conala-trainnodev.intent',
#f'{EXP_DIR}/conala-corpus/conala-dev.intent',
#f'{EXP_DIR}/conala-corpus/conala-dev.snippet',
#f'{EXP_DIR}/conala-corpus/conala-test.intent',
#f'{EXP_DIR}/conala-corpus/conala-test.snippet',
f'{EXP_DIR}/conala-corpus/attack_code_train.txt',
f'{EXP_DIR}/conala-corpus/attack_text_train.txt',
f'{EXP_DIR}/conala-corpus/attack_code_test.txt',
f'{EXP_DIR}/conala-corpus/attack_text_test.txt'
#f'{EXP_DIR}/conala-corpus/all.code',
#f'{EXP_DIR}/conala-corpus/all.anno'
],
out_files=
[ #f'{EXP_DIR}/conala-corpus/conala-trainnodev.tmspm4000.snippet',
#f'{EXP_DIR}/conala-corpus/conala-trainnodev.tmspm4000.intent',
#f'{EXP_DIR}/conala-corpus/conala-dev.tmspm4000.intent',
#f'{EXP_DIR}/conala-corpus/conala-dev.tmspm4000.snippet',
#f'{EXP_DIR}/conala-corpus/conala-test.tmspm4000.intent',
#f'{EXP_DIR}/conala-corpus/conala-test.tmspm4000.snippet',
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.snippet',
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.intent',
f'{EXP_DIR}/conala-corpus/attack-test.tmspm4000.snippet',
f'{EXP_DIR}/conala-corpus/attack-test.tmspm4000.intent'
#f'{EXP_DIR}/conala-corpus/django.tmspm4000.snippet',
#f'{EXP_DIR}/conala-corpus/django.tmspm4000.intent'
],
specs=[{
'filenum':
'all',
'tokenizers': [
SentencepieceTokenizer(
hard_vocab_limit=False,
train_files=[
f'{EXP_DIR}/conala-corpus/attack_text_train.txt',
f'{EXP_DIR}/conala-corpus/attack_code_train.txt'
],
vocab_size=self.vocab_size,
model_type=self.model_type,
model_prefix=
'conala-corpus/attack-train.tmspm4000.spm')
]
}]),
PreprocVocab(
in_files=[
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.intent',
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.snippet'
],
out_files
=[
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.intent.vocab',
f'{EXP_DIR}/conala-corpus/attack-train.tmspm4000.snippet.vocab'
],
specs=[{
'filenum':
'all',
'filters': [VocabFiltererFreq(min_freq=self.min_freq)]
}])
],
overwrite=False)
src_vocab = Vocab(
vocab_file=
f"{EXP_DIR}/conala-corpus/attack-train.tmspm4000.intent.vocab")
trg_vocab = Vocab(
vocab_file=
f"{EXP_DIR}/conala-corpus/attack-train.tmspm4000.snippet.vocab")
batcher = Batcher(batch_size=64)
inference = AutoRegressiveInference(search_strategy=BeamSearch(
len_norm=PolynomialNormalization(apply_during_search=True),
beam_size=5),
post_process='join-piece')
layer_dim = self.layer_dim
model = DefaultTranslator(
src_reader=PlainTextReader(vocab=src_vocab),
trg_reader=PlainTextReader(vocab=trg_vocab),
src_embedder=SimpleWordEmbedder(emb_dim=layer_dim,
vocab=src_vocab),
encoder=BiLSTMSeqTransducer(input_dim=layer_dim,
hidden_dim=layer_dim,
layers=self.layers),
attender=MlpAttender(hidden_dim=layer_dim,
state_dim=layer_dim,
input_dim=layer_dim),
trg_embedder=SimpleWordEmbedder(emb_dim=layer_dim,
vocab=trg_vocab),
decoder=AutoRegressiveDecoder(
input_dim=layer_dim,
rnn=UniLSTMSeqTransducer(
input_dim=layer_dim,
hidden_dim=layer_dim,
),
transform=AuxNonLinear(input_dim=layer_dim,
output_dim=layer_dim,
aux_input_dim=layer_dim),
scorer=Softmax(vocab_size=len(trg_vocab), input_dim=layer_dim),
trg_embed_dim=layer_dim,
input_feeding=False,
bridge=CopyBridge(dec_dim=layer_dim)),
inference=inference)
#decoder = AutoRegressiveDecoder(bridge=CopyBridge(),inference=inference))
train = SimpleTrainingRegimen(
name=f"{EXP}",
model=model,
batcher=WordSrcBatcher(avg_batch_size=64),
trainer=AdamTrainer(alpha=self.alpha),
patience=3,
lr_decay=0.5,
restart_trainer=True,
run_for_epochs=self.epochs,
src_file=f"{EXP_DIR}/conala-corpus/attack-train.tmspm4000.intent",
trg_file=f"{EXP_DIR}/conala-corpus/attack-train.tmspm4000.snippet",
dev_tasks=[
LossEvalTask(
src_file=
f"{EXP_DIR}/conala-corpus/attack-test.tmspm4000.intent",
ref_file=
f'{EXP_DIR}/conala-corpus/attack-test.tmspm4000.snippet',
model=model,
batcher=WordSrcBatcher(avg_batch_size=64)),
AccuracyEvalTask(
eval_metrics='bleu',
src_file=
f'{EXP_DIR}/conala-corpus/attack-test.tmspm4000.intent',
ref_file=f'{EXP_DIR}/conala-corpus/attack_text_test.txt',
hyp_file=f'results/{EXP}.dev.hyp',
model=model)
])
evaluate = [
AccuracyEvalTask(
eval_metrics="bleu",
#src_file=f"{EXP_DIR}/conala-corpus/conala-test.tmspm4000.intent",
src_file=
f"{EXP_DIR}/conala-corpus/attack-test.tmspm4000.intent",
#ref_file=f"{EXP_DIR}/conala-corpus/all.code",
#ref_file = f"{EXP_DIR}/conala-corpus/conala-test.snippet",
ref_file=f"{EXP_DIR}/conala-corpus/attack_text_test.txt",
hyp_file=f"results/{EXP}.test.hyp",
inference=inference,
model=model)
]
standard_experiment = Experiment(exp_global=ExpGlobal(
default_layer_dim=512,
dropout=0.3,
log_file=log_file,
model_file=model_file),
name="annot",
model=model,
train=train,
evaluate=evaluate)
# run experiment
standard_experiment(
save_fct=lambda: save_to_file(model_file, standard_experiment))
exit()
def test_simult_beam(self): event_trigger.set_train(False) mle_loss = loss_calculators.MLELoss() mle_loss.calc_loss(self.model, self.src[0], self.trg[0]) self.model.generate(batchers.mark_as_batch([self.src_data[0]]), BeamSearch(beam_size=2))
def test_rnng_beam(self):
event_trigger.set_train(False)
self.model.generate(batchers.mark_as_batch([self.src_data[0]]),
BeamSearch())
