在进行神经网络训练时,常会将数据分成batch进行训练,为了方便数据的加载和输入,常需要继承dataset类来重新组织自己的数据,并使用dataloader加载
- 继承dataset ,定义dataloader
#继承 Dataset类 ,将自己的数据构造成dataloader可以处理处理的形式
class DealDataset(Dataset):
def __init__(self):
self.x_data = train_X
self.y_data = train_y
self.len = train.shape[0]
def __getitem__(self, index)