def get_dataset(self, segment='train'):
"""Get the corresponding dataset for XNLI.
Parameters
----------
segment : str, default 'train'
Dataset segments. Options are 'dev', 'test', 'train'
"""
return BaiduErnieXNLI(segment)
def get_dataset(self, segment='train',
root=os.path.join(get_home_dir(), 'datasets', 'baidu_ernie_data')):
"""Get the corresponding dataset for XNLI.
Parameters
----------
segment : str, default 'train'
Dataset segments. Options are 'dev', 'test', 'train'
root : str, default $BAIDU_ERNIE_DATA_DIR/
Path to the folder which stores the dataset.
"""
return BaiduErnieXNLI(segment, root=root)