使用 GeneratorDataset 创建 TensorFlow 训练集

这是一个使用 train_dataset_generator 生成的 GeneratorDataset 类型的训练集，包含'data' 和 'label' 两个字段。shuffle=True 表示训练集在每个 epoch 开始前都会被打乱顺序。buffer_size=10 表示打乱顺序时使用的缓冲区大小为 10。batch_size=4 表示每个 batch 中包含 4 个样本，drop_remainder=True 表示如果最后一个 batch 的样本数不足 4 个，则会被丢弃。