在学习“深度学习”这门课的时候,MNIST问题被认为是深度学习的“Hello World”,为了执行这段代码(如下),需要安装一些库,比如keras。对于新手来说,安装这个库容易采坑。我把需要注意的点说一下吧。
from keras.datasets import mnist
(train_images,train_labels),(test_images,test_labels) = mnist.load_data()
train_images.shape
len(train_labels)
from keras import models
from keras import layers
network = models.Sequential()
network.add(layers.Dense(512,activation='relu',input_shape=(28*28,)))
network.add(layers.Dense(10,activation='softmax'))
network.compile(optimizer='rmsprop',loss='categorical_crossentropy',metrics=['accuracy'])
train_images = train_images.reshape((60000,28*28))
train_images = train_images.astype('float32')/255
test_images = test_images.reshape((10000,28*28))
test_images = test_images.astype('float32')/255
from keras.utils import to_categorical
train_labels = to_categorical(train_labels)
test_labels = to_categorical(test_labels)
network.fit(train_images,train_labels,epochs=5,batch_size=128)
安装keras需要注意的点:
1.在anaconda安装库时,尽量用pip,而不是conda。我刚开始用conda安装keras库之后,打开prompt总是瞬间执行许多代码(如下),有点莫名其妙,但又不知道出现的原因,也无法描述问题就不能再网上搜到解答。
2.有博主提到说,安装keras不成功的原因是开始界面的anaconda中Reset Spyder Settings没安装成功。
我试了六七个版本,每次安装完之后那个选项都不会出现,打开之后就会闪退。但这些不是不能安装keras的原因。下图是运行exe文件安装的时候就会有一个项目只安装到5%就没有继续。spyder的设置好像可以通过anaconda Navigator进行设置