MNIST 데이터셋 가져오기

[sungjk]

Problem

3장 분류에서 MNIST 데이터 로딩하는 부분에서 발생하는 에러.
mldata.org 사이트가 다운되어서 데이터를 가져오지 못함.

Solution

데이터 로딩하실 때 아래와 같이 하시면 됩니다.

from shutil import copyfileobj
from six.moves import urllib
from sklearn.datasets.base import get_data_home
import os
import ssl
ssl._create_default_https_context = ssl._create_unverified_context

def fetch_mnist(data_home=None):
    mnist_alternative_url = "https://github.com/amplab/datascience-sp14/raw/master/lab7/mldata/mnist-original.mat"
    data_home = get_data_home(data_home=data_home)
    data_home = os.path.join(data_home, 'mldata')
    if not os.path.exists(data_home):
        os.makedirs(data_home)
    mnist_save_path = os.path.join(data_home, "mnist-original.mat")
    if not os.path.exists(mnist_save_path):
        mnist_url = urllib.request.urlopen(mnist_alternative_url)
        with open(mnist_save_path, "wb") as matlab_file:
            copyfileobj(mnist_url, matlab_file)

fetch_mnist()

from sklearn.datasets import fetch_mldata
mnist = fetch_mldata("MNIST original")
mnist

Reference

scikit-learn/scikit-learn#8588

[gihojeong]

그렇게 해도 에러가 나는 것은 왜 그럴까요? 여전히 out of index 라는 에러가 발생합니다.

[sungjk]

@gihojeong 작성하신 코드와 발생한 에러 내용을 첨부해주실 수 있나요?