#data = load_dataset(data_dir=_dataset_path, name='wikitext-2-raw-v1', split=_split) data = load_dataset(os.path.join(_dataset_path,"wikitext"), name='wikitext-2-raw-v1', cache_dir=".cache", split=self._split) # Format the text column of the dataset text_list = [' \n...