Pickle incompatibility of numpy arrays between Python 2 and 3

前端 未结 7 1671
遥遥无期
遥遥无期 2020-11-28 01:12

I am trying to load the MNIST dataset linked here in Python 3.2 using this program:

import pickle
import gzip
import numpy


with gzip.open(\'mnist.pkl.gz\',         


        
7条回答
  •  臣服心动
    2020-11-28 01:53

    I just stumbled upon this snippet. Hope this helps to clarify the compatibility issue.

    import sys
    
    with gzip.open('mnist.pkl.gz', 'rb') as f:
        if sys.version_info.major > 2:
            train_set, valid_set, test_set = pickle.load(f, encoding='latin1')
        else:
            train_set, valid_set, test_set = pickle.load(f)
    

提交回复
热议问题