def get_dataset(self, segment='train'):
"""Get the corresponding dataset for MNLI
Parameters
----------
segment : str, default 'train'
Dataset segments. Options are 'dev_matched', 'dev_mismatched', 'test_matched',
'test_mismatched', 'train'
"""
return GlueMNLI(segment=segment)
def get_dataset(self, segment='train',
root=os.path.join(os.getenv('GLUE_DIR', 'glue_data'), 'MNLI')):
"""Get the corresponding dataset for MNLI
Parameters
----------
segment : str, default 'train'
Dataset segments. Options are 'dev_matched', 'dev_mismatched', 'test_matched',
'test_mismatched', 'train'
root : str, default $GLUE_DIR/MNLI
Path to the folder which stores the dataset.
"""
return GlueMNLI(segment=segment, root=root)
