ValueError:无法将大小为2322118的数组重塑为形状(60000,784)

时间:2018-02-25 16:12:59

标签: python numpy mnist

我正在使用一个图像clasiffier,它使用随机森林算法对手写的mnist数字进行分类。我正在使用从github中提取的代码。当我运行代码时,我总是得到一个说:

的错误

文件“。\ MNIST.py”,第52行,in     X = X.reshape((60000,28 * 28)) ValueError:无法将大小为2322118的数组重塑为形状(60000,784)

有人可以帮助我吗,我已经检查过stackoverflow上的类似问题,但似乎没有一个对我有用。

# coding: utf-8

# # MNIST dataset classification

import matplotlib.pyplot as plt

import os
import numpy as np
import scipy as sp

from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import StratifiedShuffleSplit
from sklearn.externals import joblib

np.random.seed(42)

# ## Fetching the data

 os.system('wget http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz')
 os.system('wget http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz')
 os.system('wget http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz')
 os.system('wget http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz')
os.system('gunzip *.gz')


# ## Loading the data to Numpy arrays


with open("train-images.idx3-ubyte", "rb") as f:
    X = np.frombuffer(f.read(), dtype=np.uint8, offset=16).copy()
    X = np.array(X)
    X = X.reshape((60000,28*28))

0 个答案:

没有答案