我正在使用一个图像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))