tatk.task.multiwoz package

Submodules

tatk.task.multiwoz.goal_generator module

class tatk.task.multiwoz.goal_generator.GoalGenerator(goal_model_path='/home/travis/build/thu-coai/tatk/data/multiwoz/goal/goal_model.pkl', corpus_path=None, boldify=False)

Bases: object

User goal generator.

__init__(goal_model_path='/home/travis/build/thu-coai/tatk/data/multiwoz/goal/goal_model.pkl', corpus_path=None, boldify=False)

Args:

goal_model_path: path to a goal model corpus_path: path to a dialog corpus to build a goal model

build_message(user_goal, boldify=<function null_boldify>)

get_user_goal()

tatk.task.multiwoz.goal_generator.do_boldify(content)

tatk.task.multiwoz.goal_generator.nomial_sample(counter: collections.Counter)

tatk.task.multiwoz.goal_generator.null_boldify(content)