def get_dataset(self, segment='train'): """Get the corresponding dataset for SST Parameters ---------- segment : str, default 'train' Dataset segments. Options are 'train', 'dev', 'test'. """ return GlueSST2(segment=segment)
def get_dataset(self, segment='train'): """Get the corresponding dataset for SST Parameters ---------- segment : str, default 'train' Dataset segments. Options are 'train', 'dev', 'test'. """ dataset = GlueSST2(segment=segment) sampler, _ = get_split_samplers(dataset, split_ratio=0.2) return SampledDataset(dataset, sampler)
def get_dataset(self, segment='train', root=os.path.join(os.getenv('GLUE_DIR', 'glue_data'), 'SST')): """Get the corresponding dataset for SST Parameters ---------- segment : str, default 'train' Dataset segments. Options are 'train', 'dev', 'test'. root : str, default $GLUE_DIR/SST Path to the folder which stores the dataset. """ return GlueSST2(segment=segment, root=root)