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)