At test time, we identify the most likely topic of the conversation based on s, and expect r i+1 to be generated from this topic.
where More specifically, we first train a topic model on (s, r) pairs from the training data.