在 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()