tatk.util.dataloader package¶
Submodules¶
tatk.util.dataloader.dataset_dataloader module¶
Dataloader base class. Every dataset should inherit this class and implement its own dataloader.
-
class
tatk.util.dataloader.dataset_dataloader.DatasetDataloader¶ Bases:
abc.ABC-
__init__()¶ Initialize self. See help(type(self)) for accurate signature.
-
abstract
load_data(*args, **kwargs)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
-
class
tatk.util.dataloader.dataset_dataloader.MultiWOZDataloader¶ Bases:
tatk.util.dataloader.dataset_dataloader.DatasetDataloader-
__init__()¶ Initialize self. See help(type(self)) for accurate signature.
-
load_data(data_dir='/home/travis/build/thu-coai/tatk/data/multiwoz', data_key='all', role='all', utterance=False, dialog_act=False, context=False, context_window_size=0, context_dialog_act=False, belief_state=False, last_opponent_utterance=False, last_self_utterance=False, ontology=False, session_id=False, span_info=False, terminated=False, goal=False)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
tatk.util.dataloader.module_dataloader module¶
-
class
tatk.util.dataloader.module_dataloader.ActPolicyDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.ActUserPolicyDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.AgentDSTDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.ModuleDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
abc.ABC-
__init__(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Initialize self. See help(type(self)) for accurate signature.
-
abstract
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.MultiTurnNLGDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.MultiTurnNLUDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.SingleTurnNLGDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.SingleTurnNLUDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.UserDSTDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
tatk.util.dataloader.module_dataloader.WordPolicyDataloader(dataset_dataloader: tatk.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
tatk.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-