tensorflow2.0 Mnist数据集快速读取(Python深度学习第2章)

环境: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)

 

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值