Estoy tratando de cargar el conjunto de datos MNIST vinculado aquí en Python 3.2 usando este programa:
import pickle
import gzip
import numpy
with gzip.open('mnist.pkl.gz', 'rb') as f:
l = list(pickle.load(f))
print(l)
Lamentablemente, me da el error:
Traceback (most recent call last):
File "mnist.py", line 7, in <module>
train_set, valid_set, test_set = pickle.load(f)
UnicodeDecodeError: 'ascii' codec can't decode byte 0x90 in position 614: ordinal not in range(128)
Luego intenté decodificar el archivo en escabeche en Python 2.7 y volver a codificarlo. Entonces, ejecuté este programa en Python 2.7:
import pickle
import gzip
import numpy
with gzip.open('mnist.pkl.gz', 'rb') as f:
train_set, valid_set, test_set = pickle.load(f)
# Printing out the three objects reveals that they are
# all pairs containing numpy arrays.
with gzip.open('mnistx.pkl.gz', 'wb') as g:
pickle.dump(
(train_set, valid_set, test_set),
g,
protocol=2) # I also tried protocol 0.
Se ejecutó sin error, así que volví a ejecutar este programa en Python 3.2:
import pickle
import gzip
import numpy
# note the filename change
with gzip.open('mnistx.pkl.gz', 'rb') as f:
l = list(pickle.load(f))
print(l)
Sin embargo, me dio el mismo error que antes. ¿Cómo hago para que esto funcione?
Este es un mejor enfoque para cargar el conjunto de datos MNIST.