Как отделить входные данные и цели от Pytorch Fashion MNIST?

Question

Набор данных Fashion MNIST довольно странно реализован в Pytorch. Я хочу сделать что-то вроде:

X, y = FashionMNIST

Но на самом деле все немного сложнее. Вот что у меня есть:

from torchvision.datasets import FashionMNIST
train = FashionMNIST(root='.', download=True, train=True)
print(train)

Вывод:

Dataset FashionMNIST
    Number of datapoints: 60000
    Root location: c:/users/nicolas/documents/data/fashionmnist
    Split: Train

На что один наблюдение выглядит так:

print(train[0])

(<PIL.Image.Image image mode=L size=28x28 at 0x20868074780>, 9)

Я мог сделать это только для одного наблюдения.

X, y = train[0]

Итак, как мне отделить вход и цели?

Answer 1

FashionMNIST объект имеет data и targets атрибуты.

Вы можете просто написать

X, y = train.data, train.targets

, а затем вы можете увидеть формы

X.shape, y.shape

(Размер горелки ([60000, 28, 28]), Размер горелки ([60000]))