为keras中的训练生成数据

from keras.utils import Sequence

class DataGenerator(Sequence):
    def __init__(self, data, labels, batch_size):
        self.data = data
        self.labels = labels
        self.batch_size = batch_size

    def __getitem__(self, index):
        batch_data = self.data[index * self.batch_size : (index + 1) * self.batch_size]
        batch_labels = self.labels[index * self.batch_size : (index + 1) * self.batch_size]
        # 在这里进行数据预处理或增强操作
        return batch_data, batch_labels

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

# 使用数据生成器进行模型训练
train_data = ...
train_labels = ...
batch_size = 32
generator = DataGenerator(train_data, train_labels, batch_size)
model.fit(generator, epochs=10)