dialogue#

class DialogueTask(task, model, batch_size=1, max_seq_len=512, **kwargs)[source]#

Bases: Task

Task of Chinese open domain dialogue. :param task: The name of task. :type task: string :param model: The model name in the task. :type model: string :param kwargs: Additional keyword arguments passed along to the specific task. :type kwargs: dict, optional

interactive_mode(max_turn=3)[source]#

Enter the interactive mode.