mahdiabdollahpour/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.

Python