Cannot get MNIST database through Anaconda/jupyter

前端 未结 9 2590
情歌与酒
情歌与酒 2021-02-13 14:20

Hu guys,

I\'m new to python/anaconda/jupyter/numPy, panda, etc.... so please excuse me if it\'s a really stupid question. I\'m trying to obtain MNIST database by using a

9条回答
  •  天涯浪人
    2021-02-13 14:46

    from sklearn.datasets import fetch_mldata
    try:
        mnist = fetch_mldata('MNIST original')
    except Exception as ex:        
        from six.moves import urllib
        from scipy.io import loadmat
        import os
    
        mnist_path = os.path.join(".", "datasets", "mnist-original.mat")
    
        # download dataset from github.
        mnist_alternative_url = "https://github.com/amplab/datascience-sp14/raw/master/lab7/mldata/mnist-original.mat"
        response = urllib.request.urlopen(mnist_alternative_url)
        with open(mnist_path, "wb") as f:
            content = response.read()
            f.write(content)
    
        mnist_raw = loadmat(mnist_path)
        mnist = {
            "data": mnist_raw["data"].T,
            "target": mnist_raw["label"][0],
            "COL_NAMES": ["label", "data"],
            "DESCR": "mldata.org dataset: mnist-original",
        }
        print("Done!")
    

提交回复
热议问题