tatk.policy.vector package

Submodules

tatk.policy.vector.dataset module

class tatk.policy.vector.dataset.ActDataset(s_s, a_s)

Bases: torch.utils.data.dataset.Dataset

__init__(s_s, a_s)

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

class tatk.policy.vector.dataset.ActStateDataset(s_s, a_s, next_s)

Bases: torch.utils.data.dataset.Dataset

__init__(s_s, a_s, next_s)

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

tatk.policy.vector.vector_camrest module

class tatk.policy.vector.vector_camrest.CamrestVector(voc_file, voc_opp_file, character='sys', intent_file='/home/travis/build/thu-coai/tatk/data/camrest/trackable_intent.json')

Bases: tatk.policy.vec.Vector

__init__(voc_file, voc_opp_file, character='sys', intent_file='/home/travis/build/thu-coai/tatk/data/camrest/trackable_intent.json')

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

action_devectorize(action_vec)

recover an action

Args:

action_vec (np.array):

Dialog act vector

Returns:

action (tuple): Dialog act

action_vectorize(action)

generate_dict()

init the dict for mapping state/action into vector

one_hot_vector(num)

Return number of available entities for particular domain.

pointer(turn)

state_vectorize(state)

vectorize a state

Args:

state (dict):

Dialog state

action (tuple):

Dialog act

Returns:

state_vec (np.array): Dialog state vector

tatk.policy.vector.vector_crosswoz module

class tatk.policy.vector.vector_crosswoz.CrossWozVector(sys_da_voc_json, usr_da_voc_json)

Bases: tatk.policy.vec.Vector

__init__(sys_da_voc_json, usr_da_voc_json)

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

action_devectorize(action_vec)

must call state_vectorize func before :param action_vec: :return:

action_vectorize(da)

generate_dict()

init the dict for mapping state/action into vector

state_vectorize(state)

vectorize a state

Args:

state (tuple):

Dialog state

Returns:

state_vec (np.array):

Dialog state vector

tatk.policy.vector.vector_multiwoz module

class tatk.policy.vector.vector_multiwoz.MultiWozVector(voc_file, voc_opp_file, character='sys', intent_file='/home/travis/build/thu-coai/tatk/data/multiwoz/trackable_intent.json')

Bases: tatk.policy.vec.Vector

__init__(voc_file, voc_opp_file, character='sys', intent_file='/home/travis/build/thu-coai/tatk/data/multiwoz/trackable_intent.json')

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

action_devectorize(action_vec)

recover an action Args:

action_vec (np.array):

Dialog act vector

Returns:

action (tuple):

Dialog act

action_vectorize(action)

generate_dict()

init the dict for mapping state/action into vector

one_hot_vector(num, domain, vector)

Return number of available entities for particular domain.

pointer(turn)

state_vectorize(state)

vectorize a state

Args:

state (dict):

Dialog state

action (tuple):

Dialog act

Returns:

state_vec (np.array):

Dialog state vector