在MXNet中加載和處理數據集通常需要使用DataLoader類和Dataset類。
import mxnet as mx
from mxnet.gluon.data.vision import datasets
train_data = datasets.MNIST(train=True)
test_data = datasets.MNIST(train=False)
from mxnet.gluon.data.vision import transforms
transformer = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(0.13, 0.31)
])
train_data = train_data.transform_first(transformer)
test_data = test_data.transform_first(transformer)
train_loader = mx.gluon.data.DataLoader(train_data, batch_size=64, shuffle=True)
通過以上步驟，就可以加載和處理數據集并創建DataLoader來批量加載數據用于模型訓練。
