/beam-search-with-rollouts

Implementing Simulation-guided Beam Search for text generation. Idea already proposed in Choo et al for Neural Combinatorial Optimization. In this repo, we use the technique for Controllable Text Generation.

Primary LanguagePython

Stargazers