tatk.task.camrest package

Submodules

tatk.task.camrest.goal_generator module

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

Bases: object

User goal generator.

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

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)

get_user_goal()

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