环境:tensorflow版本:2.0.0b0
问题:在Python深度学习第二章中,从Keras加载Mnist数据集,由于某些未知的原因,导致数据集加载慢,甚至会被当成爬虫,而终止访问。
import tensorflow as tf
import tensorflow.keras as keras
(train_images,train_labels),(test_images,test_labels)=keras.datasets.mnist.load_data()
将上述代码代替《Python深度学习》P21页的代码清单2-1 加载Keras中的MNIST数据集的代码,即可快速下载数据集。
本章完整例子代码如下:
import tensorflow as tf
import tensorflow.keras as keras
import matplotlib.pyplot as plt
from keras import models
from keras import layers
from keras.utils import to_categorical
(train_images,train_labels),(test_images,test_labels)=keras.datasets.mnist.load_data()
network = models.Sequential()
network.add(layers.Dense(812,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
train_labels = to_categorical(train_labels)
test_labels = to_categorical(test_labels)
network.fit(train_images,train_labels,epochs=5,batch_size=128)
test_loss,test_acc = network.evaluate(test_images,test_labels)
print('test_acc:',test_acc)