使 Dataset 类迭代所有文件

如何解决使 Dataset 类迭代所有文件

我有下面的数据集类，它从列表中检索一个项目并提取一些特征。
如何更改它以使其遍历文件列表中的每个项目并返回所有项目的特征，而不是仅返回索引的特征？

class MyDataset(Dataset):
    def __init__(self,data_dir,transform=None):    
        self.data_dir = data_dir
        self.file_lst = sorted([os.path.join(path,file) for path,dirs,files in os.walk('/folder/folder/folder') for file in files if fnmatch.fnmatch(file,'*wav')])
        self.transform = transforms.Compose(transform)
    
    def __len__(self):
        return len(self.file_lst)
    
    def __getitem__(self,idx):
        return self.transform(self.file_lst[idx])

train_dataset = MyDataset(input_dir,[file_reader,feat_extractor])
train_dataset[0]