tatk.policy.vhus.multiwoz package

Submodules

tatk.policy.vhus.multiwoz.train module

@author: truthless

tatk.policy.vhus.multiwoz.usermanager module

@author: keshuichonglx

class tatk.policy.vhus.multiwoz.usermanager.UserDataManager

Bases: object

__init__()

Initialize self. See help(type(self)) for accurate signature.

data_loader()

data_loader_seg()

get_goal_id(goal)

get_sysda_id(sys_das)

get_usrda_id(usr_das)

get_voc_size()

id2sentence(ids)

org_data_loader()

static query_goal_for_sys(domain, slot, value, goal)

static query_goal_for_usr(domain, slot, value, goal)

static ref_data2stand(da)

static sysda2seq(sys_da: dict, goal: dict)

static train_test_val_split(goals, usr_dass, sys_dass, test_size=0.1, val_size=0.1)

static train_test_val_split_seg(goals_seg, usr_dass_seg, sys_dass_seg, test_size=0.1, val_size=0.1)

static usrda2seq(usr_da: dict, goal: dict)

static usrgoal2seq(goal: dict)

static usrseq2da(usr_seq: list, goal: dict)

vocab_loader()

tatk.policy.vhus.multiwoz.vhus module

class tatk.policy.vhus.multiwoz.vhus.UserPolicyVHUS(archive_file='/home/travis/build/thu-coai/tatk/tatk/policy/vhus/multiwoz/models/vhus_simulator_multiwoz.zip', model_file='https://tatk-data.s3-ap-northeast-1.amazonaws.com/vhus_simulator_multiwoz.zip')

Bases: tatk.policy.vhus.vhus.UserPolicyVHUSAbstract

__init__(archive_file='/home/travis/build/thu-coai/tatk/tatk/policy/vhus/multiwoz/models/vhus_simulator_multiwoz.zip', model_file='https://tatk-data.s3-ap-northeast-1.amazonaws.com/vhus_simulator_multiwoz.zip')

Initialize self. See help(type(self)) for accurate signature.

predict(state)

Predict an user act based on state and preorder system action.

Args:

state (tuple):

Dialog state.

Returns:

usr_action (tuple):

User act.

session_over (boolean):

True to terminate session, otherwise session continues.