scikit-learn使用fetch_mldata无法下载MNIST数据集问题解决方法

一.问题描述

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

使用scikit-learn下载MNIST数据集的时候报错。如下图所示：

二.问题分析

1.Python版本为 ：Python 3.7.1 (default, Dec 10 2018, 22:54:23) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.

三.解决方法

1.直接下载MNST数据集文件。

https://github.com/amplab/datascience-sp14/raw/master/lab7/mldata/mnist-original.mat

2.将下载的数据集文件放到scikit数据根目录下的mldata目录。

根目录查询方式：

from sklearn.datasets.base import get_data_home 
print (get_data_home()) # 如我的电脑上的目录为： C:\Users\95232\scikit_learn_data

如：根据get_data_home() 查询到我电脑上的scikit data home目录为：

C:\Users\95232\scikit_learn_data

那么我需要把下载的MNST数据集文件放置到目录：

C:\Users\95232\scikit_learn_data\mldata

 

四.参考资料

[1] https://github.com/ageron/handson-ml/issues/143