out_file.flush()
if opt.verbose:
words = []
for f in src_sent:
word = translator.fields["src"].vocab.itos[f]
if word == onmt.IO.PAD_WORD:
break
words.append(word)
os.write(1, bytes("\nSENT %d: %s\n" %
(count, " ".join(words)), "UTF-8"))
index += 1
os.write(1, bytes("PRED %d: %s\n" %
After Change
(sent.squeeze(1) for sent in src.split(1, dim=1)))
for pred_sents, gold_sent, pred_score, gold_score, src_sent in z_batch:
n_best_preds = [" ".join(pred) for pred in pred_sents[:opt.n_best]]
count += 1
out_file.write("\n".join(n_best_preds))
out_file.write("\n")
out_file.flush()
if opt.verbose:
words = get_src_words(
src_sent, translator.fields["src"].vocab.itos)
os.write(1, bytes("\nSENT %d: %s\n" %
(count, words), "UTF-8"))
index += 1
best_pred = n_best_preds[0]
best_score = pred_score[0]
os.write(1, bytes("PRED %d: %s\n" %
(count, best_pred), "UTF-8"))
print("PRED SCORE: %.4f" % best_score)
