63871d5ab3301d015805abddd8f4259680d6a24a,translate.py,,main,#,68

Before Change

            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

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)

Instances