convlab2.util.dataloader package¶
Submodules¶
convlab2.util.dataloader.dataset_dataloader module¶
Dataloader base class. Every dataset should inherit this class and implement its own dataloader.
-
class
convlab2.util.dataloader.dataset_dataloader.CamrestDataloader¶ Bases:
convlab2.util.dataloader.dataset_dataloader.DatasetDataloader-
load_data(data_dir='/home/travis/build/thu-coai/ConvLab-2/data/camrest', data_key='all', role='all', utterance=False, dialog_act=False, context=False, context_window_size=0, context_dialog_act=False, last_opponent_utterance=False, last_self_utterance=False, session_id=False, terminated=False, goal=False)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
-
class
convlab2.util.dataloader.dataset_dataloader.CrossWOZDataloader(en=False)¶ Bases:
convlab2.util.dataloader.dataset_dataloader.DatasetDataloader-
load_data(data_dir=None, data_key='all', role='all', utterance=False, dialog_act=False, context=False, context_window_size=0, context_dialog_act=False, user_state=False, sys_state=False, sys_state_init=False, last_opponent_utterance=False, last_self_utterance=False, session_id=False, terminated=False, goal=False, final_goal=False, task_description=False)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
-
class
convlab2.util.dataloader.dataset_dataloader.DatasetDataloader¶ Bases:
abc.ABC-
abstract
load_data(*args, **kwargs)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
abstract
-
class
convlab2.util.dataloader.dataset_dataloader.DealOrNotDataloader¶ Bases:
convlab2.util.dataloader.dataset_dataloader.DatasetDataloader-
load_data(data_dir='/home/travis/build/thu-coai/ConvLab-2/data/deal_or_not', data_key='all', role='all', utterance=False, context=False, context_window_size=0, last_opponent_utterance=False, last_self_utterance=False, session_id=False, terminated=False, goal=False, output=False, partner_input=False)¶ load data from file, according to what is need :param args: :param kwargs: :return: data
-
-
class
convlab2.util.dataloader.dataset_dataloader.MultiWOZDataloader(zh=False)¶ Bases:
convlab2.util.dataloader.dataset_dataloader.DatasetDataloader-
load_data(data_dir=None, 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
-
convlab2.util.dataloader.module_dataloader module¶
-
class
convlab2.util.dataloader.module_dataloader.ActPolicyDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.ActUserPolicyDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.AgentDSTDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.CrossWOZAgentDSTDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.ModuleDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
abc.ABC-
abstract
load_data(*args, **kwargs)¶
-
abstract
-
class
convlab2.util.dataloader.module_dataloader.MultiTurnNLGDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.MultiTurnNLUDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.SingleTurnNLGDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.SingleTurnNLUDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.UserDSTDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-
-
class
convlab2.util.dataloader.module_dataloader.WordPolicyDataloader(dataset_dataloader: convlab2.util.dataloader.dataset_dataloader.DatasetDataloader)¶ Bases:
convlab2.util.dataloader.module_dataloader.ModuleDataloader-
load_data(*args, **kwargs)¶
-