c2d5bde2b2d100b5d11e2c7f2e58186bca296f0d,src/beamsearch.py,,write_translation_preds,#Any#Any#Any#Any#,245

Before Change

    with open(preds_file_path, "a") as f:
        for i in range(len(hyps)):
            hyp, target = hyps[i], relevant_targets[i]
            hyp_sentence = " ".join([_get_word(decoder_vocab, i) for i in hyp])
            target_sentence = " ".join([_get_word(decoder_vocab, i) for i in target])
            f.write("{}\t{}\n".format(hyp_sentence, target_sentence))

After Change

    with open(preds_file_path, "a") as f:
        for i in range(len(hyps)):
            // form hyp sentence
            hyp_sentence = []
            for word_idx in hyps[i, 1:]:
                hyp_sentence.append(
                    _get_word(decoder_vocab, word_idx)
                )
                if word_idx == decoder_vocab.get_token_index(END_SYMBOL, "targets"):
                    break
            hyp_sentence = " ".join(hyp_sentence)

            target = relevant_targets[i]
            target_sentence = " ".join([_get_word(decoder_vocab, i) for i in target])
            f.write("{}\t{}\n".format(hyp_sentence, target_sentence))

Instances