从Pytorch dataset对象中删除样本

import torch
from torch.utils.data import Dataset

class CustomDataset(Dataset):
    def __init__(self, data):
        self.data = data

    def __getitem__(self, index):
        return self.data[index]

    def __len__(self):
        return len(self.data)

# 创建自定义Dataset对象
data = [1, 2, 3, 4, 5]
dataset = CustomDataset(data)

# 方法一：使用索引删除样本
index_to_delete = 2
deleted_sample = dataset[index_to_delete]  # 获取要删除的样本
del dataset[index_to_delete]  # 删除样本
print("Deleted sample:", deleted_sample)
print("Updated dataset:", dataset.data)

# 方法二：使用过滤器删除样本
filter_condition = lambda x: x % 2 == 0  # 删除偶数样本
indices_to_delete = [i for i, sample in enumerate(dataset) if filter_condition(sample)]  # 筛选要删除的样本索引
deleted_samples = [dataset[i] for i in indices_to_delete]  # 获取要删除的样本
for i in sorted(indices_to_delete, reverse=True):  # 倒序删除样本
    del dataset[i]
print("Deleted samples:", deleted_samples)
print("Updated dataset:", dataset.data)