e185a7110ec79864dcef5114b71356691eec39df,theanolm/commands/sample.py,,sample,#,37

Before Change

    sys.stdout.flush()
    sampler = TextSampler(network)

    for i in range(args.num_sentences):
        words = sampler.generate()
        args.output_file.write("{}: {}\n".format(
            i, " ".join(words)))

After Change

    sys.stdout.flush()
    sampler = TextSampler(network)

    sequences = sampler.generate(30, args.num_sentences)
    for sequence in sequences:
        try:
            eos_pos = sequence.index("</s>")
            sequence = sequence[:eos_pos+1]

Instances