MNISTデータセットがダウンロードできないバグの修正

[KotaShimomura]

概要

第9回 MLPによる多クラス分類（MNIST）

上記のnotebookでMNISTがダウンロードできません． そのため，torchvisionのMNISTを利用する方法に変更しています．

変更点

• 該当notebookの編集 MNISTのダウンロード，読み込み部分のプログラムを変更．

  
  import torchvision
  train_data = torchvision.datasets.MNIST(root="./", train=True, download=True)
  test_data = torchvision.datasets.MNIST(root="./", train=False, download=True)

train_data = torchvision.datasets.MNIST(root="./", train=True, download=True) test_data = torchvision.datasets.MNIST(root="./", train=False, download=True)

x_train = train_data.data.numpy().reshape(-1, 784) y_train = train_data.targets.numpy() x_test = test_data.data.numpy().reshape(-1, 784) y_test = test_data.targets.numpy()

print(x_train.shape, y_train.shape) print(x_test.shape, y_test.shape)