train_dataset = ds.NumpySlicesDataset((X_train, y_train), shuffle=True) train_dataset = train_dataset.batch(batch_size, drop_remainder=True)
时间: 2023-12-06 09:05:26 浏览: 90
train_data-数据集
5星 · 资源好评率100%
这段代码是使用MindSpore框架中的数据集(ds)模块,将训练数据(X_train和y_train)转换为NumpySlicesDataset类型的数据集。然后,使用batch()函数将数据集按照batch_size进行切分,并丢掉最后一组不足batch_size的数据。最后,shuffle=True表示在每个epoch开始之前将数据集打乱顺序。
阅读全文