在Python 3中加载mnist数据集时出错

Question

当尝试使用mnist_loader脚本加载MNIST数据集时，我不断遇到以下错误： “ ascii”编解码器无法解码位置614的字节0x90：序数不在范围（128）中

用于加载数据的代码如下所示。有什么建议么？

非常感谢！

def load_data():
    f = gzip.open('mnist.pkl.gz', 'rb')
    training_data, validation_data, test_data = pickle.load(f, encoding="latin1")
    f.close()
    return (training_data, validation_data, test_data)

Answer 1

这可能适用于Python 3，代码如下：

import pickle
import gzip
with gzip.open('mnist.pkl.gz', 'rb') as f:
    data = pickle._Unpickler(f)
    data.encoding = 'latin1'  # set encoding
    train, valid, test = data.load()