tf.keras学习笔记(一) mnist分类

这篇博客是tf.keras学习系列的第一部分,主要介绍了如何使用tf.keras进行MNIST手写数字数据集的分类。通过导入keras库并展示简单的代码实现,博主展示了模型训练和运行的基本流程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

tf.keras学习笔记(一)

  • mnist分类

导入keras的包

from tensorflow.contrib import keras

另外一些常见的数据集keras.datasets提供了
在这里插入图片描述

  • 代码:
import tensorflow as tf
# import keras form tf
from tensorflow.contrib import keras

# load mnist
mnist = keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

# build model
model = keras.models.Sequential()
model.add(keras.layers.Flatten(input_shape=(28,28)))
model.add(keras.layers.Dense(512, activation=tf.nn.leaky_relu))
model.add(keras.layers.Dropout(0.2))
model.add(keras.layers.Dense(10,activation=tf.nn.softmax))

# define optimizer and loss
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# fit the training data
model.fit(x_train, y_train, epochs=5)

# test the test data
model.evaluate(x_test, y_test)

  • 运行结果:
/usr/bin/python3.6 /media/todd/38714CA0C89E958E/147/yl_tmp/readingbook/pipeline/gesture_reco/gesture_reco.py
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
    8192/11490434 [..............................] - ETA: 0s
   57344/11490434 [..............................] - ETA: 28s
  163840/11490434 [..............................] - ETA: 13s
  221184/11490434 [..............................] - ETA: 16s
  442368/11490434 [>.............................] - ETA: 9s 
  573440/11490434 [>.............................] - ETA: 8s
  655360/11490434 [>.............................] - ETA: 8s
 1212416/11490434 [==>...........................] - ETA: 4s
 1572864/11490434 [===>..........................] - ETA: 3s
 1785856/11490434 [===>..........................] - ETA: 4s
 2490368/11490434 [=====>........................] - ETA: 2s
 3571712/11490434 [========>.....................] - ETA: 2s
 3588096/11490434 [========>.....................] - ETA: 2s
 3604480/11490434 [========>.....................] - ETA: 2s
 4145152/11490434 [=========>....................] - ETA: 2s
 4194304/11490434 [=========>....................] - ETA: 2s
 6619136/11490434 [================>.............] - ETA: 0s
 6635520/11490434 [================>.............] - ETA: 1s
 6684672/11490434 [================>.............] - ETA: 1s
 6832128/11490434 [================>.............] - ETA: 1s
 8028160/11490434 [===================>..........] - ETA: 0s
 9404416/11490434 [=======================>......] - ETA: 0s
 9977856/11490434 [=========================>....] - ETA: 0s
10092544/11490434 [=========================>....] - ETA: 0s
10797056/11490434 [===========================>..] - ETA: 0s
11493376/11490434 [==============================] - 2s 0us/step
Epoch 1/5
   32/60000 [..............................] - ETA: 2:50:32 - loss: 2.3859 - acc: 0.0938
  256/60000 [..............................] - ETA: 21:26 - loss: 1.9522 - acc: 0.3477  
  576/60000 [..............................] - ETA: 9:34 - loss: 1.4732 - acc: 0.5573 
  864/60000 [..............................] - ETA: 6:24 - loss: 1.2103 - acc: 0.6366
 1120/60000 [..............................] - ETA: 4:58 - loss: 1.0855 - acc: 0.6687
 1408/60000 [..............................] - ETA: 3:58 - loss: 0.9779 - acc: 0.6960
 1728/60000 [..............................] - ETA: 3:14 - loss: 0.8756 - acc: 0.7251
 2016/60000 [>.............................] - ETA: 2:47 - loss: 0.8299 - acc: 0.7391
 2272/60000 [>.............................] - ETA: 2:29 - loss: 0.7826 - acc: 0.7544
 2560/60000 [>.............................] - ETA: 2:13 - loss: 0.7444 - acc: 0.7680
 2880/60000 [>.............................] - ETA: 1:58 - loss: 0.7013 - acc: 0.7812
 3232/60000 [>.............................] - ETA: 1:46 - loss: 0.6679 - acc: 0.7915
 3552/60000 [>.............................] - ETA: 1:36 - loss: 0.6362 - acc: 0.8029
 3872/60000 [>.............................] - ETA: 1:29 - loss: 0.6127 - acc: 0.8102
 4160/60000 [=>............................] - ETA: 1:23 - loss: 0.5971 - acc: 0.8161
 4512/60000 [=>............................] - ETA: 1:16 - loss: 0.5778 - acc: 0.8225
 4800/60000 [=>............................] - ETA: 1:12 - loss: 0.5588 - acc: 0.8290
 5056/60000 [=>............................] - ETA: 1:09 - loss: 0.5496 - acc: 0.8319
 5376/60000 [=>............................] - ETA: 1:05 - loss: 0.5360 - acc: 0.8358
 5760/60000 [=>............................] - ETA: 1:00 - loss: 0.5275 - acc: 0.8398
 6080/60000 [==>...........................] - ETA: 57s - loss: 0.5181 - acc: 0.8426 
 6400/60000 [==>...........................] - ETA: 55s - loss: 0.5122 - acc: 0.8442
 6752/60000 [==>...........................] - ETA: 52s - loss: 0.5036 - acc: 0.8479
 7072/60000 [==>...........................] - ETA: 50s - loss: 0.4966 - acc: 0.8498
 7360/60000 [==>...........................] - ETA: 48s - loss: 0.4901 - acc: 0.8519
 7584/60000 [==>...........................] - ETA: 47s - loss: 0.4856 - acc: 0.8536
 7808/60000 [==>...........................] - ETA: 45s - loss: 0.4809 - acc: 0.8554
 8128/60000 [===>..........................] - ETA: 44s - loss: 0.4762 - acc: 0.8575
 8416/60000 [===>..........................] - ETA: 42s - loss: 0.4718 - acc: 0.8591
 8640/60000 [===>..........................] - ETA: 41s - loss: 0.4657 - acc: 0.8604
 8832/60000 [===>..........................] - ETA: 40s - loss: 0.4617 - acc: 0.8611
 9024/60000 [===>..........................] - ETA: 40s - loss: 0.4580 - acc: 0.8623
 9280/60000 [===>..........................] - ETA: 39s - loss: 0.4538 - acc: 0.8636
 9504/60000 [===>..........................] - ETA: 38s - loss: 0.4521 - acc: 0.8641
 9792/60000 [===>..........................] - ETA: 37s - loss: 0.4466 - acc: 0.8658
10016/60000 [====>.........................] - ETA: 36s - loss: 0.4443 - acc: 0.8663
10304/60000 [====>.........................] - ETA: 35s - loss: 0.4387 - acc: 0.8679
10560/60000 [====>.........................] - ETA: 34s - loss: 0.4344 - acc: 0.8697
10848/60000 [====>.........................] - ETA: 33s - loss: 0.4306 - acc: 0.8708
11168/60000 [====>.........................] - ETA: 32s - loss: 0.4278 - acc: 0.8718
11424/60000 [====>.........................] - ETA: 32s - loss: 0.4243 - acc: 0.8730
11680/60000 [====>.........................] - ETA: 31s - loss: 0.4214 - acc: 0.8736
11904/60000 [====>.........................] - ETA: 31s - loss: 0.4178 - acc: 0.8747
12128/60000 [=====>........................] - ETA: 30s - loss: 0.4165 - acc: 0.8756
12416/60000 [=====>........................] - ETA: 29s - loss: 0.4118 - acc: 0.8769
12640/60000 [=====>........................] - ETA: 29s - loss: 0.4084 - acc: 0.8777
12896/60000 [=====>........................] - ETA: 28s - loss: 0.4040 - acc: 0.8790
13216/60000 [=====>........................] - ETA: 28s - loss: 0.4011 - acc: 0.8801
13504/60000 [=====>........................] - ETA: 27s - loss: 0.3982 - acc: 0.8809
13728/60000 [=====>........................] - ETA: 27s - loss: 0.3977 - acc: 0.8813
13984/60000 [=====>........................] - ETA: 26s - loss: 0.3959 - acc: 0.8821
14240/60000 [======>.......................] - ETA: 26s - loss: 0.3956 - acc: 0.8822
14464/60000 [======>.......................] - ETA: 25s - loss: 0.3940 - acc: 0.8822
14720/60000 [======>.......................] - ETA: 25s - loss: 0.3923 - acc: 0.8827
14944/60000 [======>.......................] - ETA: 25s - loss: 0.3910 - acc: 0.8830
15200/60000 [======>.......................] - ETA: 24s - loss: 0.3881 - acc: 0.8838
15456/60000 [======>.......................] - ETA: 24s - loss: 0.3872 - acc: 0.8837
15680/60000 [======>.......................] - ETA: 24s - loss: 0.3862 - acc: 0.8842
15872/60000 [======>.......................] - ETA: 23s - loss: 0.3853 - acc: 0.8846
16128/60000 [=======>......................] - ETA: 23s - loss: 0.3836 - acc: 0.8852
16416/60000 [=======>......................] - ETA: 23s - loss: 0.3806 - acc: 0.8862
16608/60000 [=======>......................] - ETA: 22s - loss: 0.3793 - acc: 0.8866
16864/60000 [=======>......................] - ETA: 22s - loss: 0.3775 - acc: 0.8871
17120/60000 [=======>......................] - ETA: 22s - loss: 0.3757 - acc: 0.8876
17344/60000 [=======>......................] - ETA: 21s - loss: 0.3748 - acc: 0.8879
17600/60000 [=======>......................] - ETA: 21s - loss: 0.3732 - acc: 0.8884
17824/60000 [=======>......................] - ETA: 21s - loss: 0.3717 - acc: 0.8887
18112/60000 [========>.....................] - ETA: 20s - loss: 0.3704 - acc: 0.8891
18400/60000 [========>.....................] - ETA: 20s - loss: 0.3684 - acc: 0.8898
18656/60000 [========>.....................] - ETA: 20s - loss: 0.3676 - acc: 0.8899
18848/60000 [========>.....................] - ETA: 20s - loss: 0.3658 - acc: 0.8905
19072/60000 [========>.....................] - ETA: 19s - loss: 0.3642 - acc: 0.8911
19264/60000 [========>.....................] - ETA: 19s - loss: 0.3634 - acc: 0.8916
19488/60000 [========>.....................] - ETA: 19s - loss: 0.3619 - acc: 0.8919
19744/60000 [========>.....................] - ETA: 19s - loss: 0.3604 - acc: 0.8924
19936/60000 [========>.....................] - ETA: 19s - loss: 0.3592 - acc: 0.8928
20192/60000 [=========>....................] - ETA: 18s - loss: 0.3578 - acc: 0.8937
20416/60000 [=========>....................] - ETA: 18s - loss: 0.3556 - acc: 0.8942
20640/60000 [=========>....................] - ETA: 18s - loss: 0.3546 - acc: 0.8944
20928/60000 [=========>....................] - ETA: 18s - loss: 0.3536 - acc: 0.8947
21216/60000 [=========>....................] - ETA: 17s - loss: 0.3518 - acc: 0.8951
21472/60000 [=========>....................] - ETA: 17s - loss: 0.3498 - acc: 0.8957
21728/60000 [=========>....................] - ETA: 17s - loss: 0.3479 - acc: 0.8961
21984/60000 [=========>....................] - ETA: 17s - loss: 0.3465 - acc: 0.8964
22272/60000 [==========>...................] - ETA: 16s - loss: 0.3455 - acc: 0.8969
22528/60000 [==========>...................] - ETA: 16s - loss: 0.3443 - acc: 0.8974
22816/60000 [==========>...................] - ETA: 16s - loss: 0.3432 - acc: 0.8977
23072/60000 [==========>...................] - ETA: 16s - loss: 0.3420 - acc: 0.8980
23328/60000 [==========>...................] - ETA: 15s - loss: 0.3411 - acc: 0.8982
23584/60000 [==========>...................] - ETA: 15s - loss: 0.3390 - acc: 0.8989
23840/60000 [==========>...................] - ETA: 15s - loss: 0.3376 - acc: 0.8993
24096/60000 [===========>..................] - ETA: 15s - loss: 0.3367 - acc: 0.8997
24384/60000 [===========>..................] - ETA: 15s - loss: 0.3355 - acc: 0.9001
24640/60000 [===========>..................] - ETA: 14s - loss: 0.3342 - acc: 0.9004
24928/60000 [===========>..................] - ETA: 14s - loss: 0.3325 - acc: 0.9011
25184/60000 [===========>..................] - ETA: 14s - loss: 0.3315 - acc: 0.9013
25472/60000 [===========>..................] - ETA: 14s - loss: 0.3296 - acc: 0.9016
25728/60000 [===========>..................] - ETA: 14s - loss: 0.3288 - acc: 0.9018
25984/60000 [===========>..................] - ETA: 14s - loss: 0.3273 - acc: 0.9021
26240/60000 [============>.................] - ETA: 13s - loss: 0.3261 - acc: 0.9026
26496/60000 [============>.................] - ETA: 13s - loss: 0.3248 - acc: 0.9030
26688/60000 [============>.................] - ETA: 13s - loss: 0.3244 - acc: 0.9031
26912/60000 [============>.................] - ETA: 13s - loss: 0.3233 - acc: 0.9036
27168/60000 [============>.................] - ETA: 13s - loss: 0.3224 - acc: 0.9039
27488/60000 [============>.................] - ETA: 13s - loss: 0.3219 - acc: 0.9040
27712/60000 [============>.................] - ETA: 12s - loss: 0.3213 - acc: 0.9040
27968/60000 [============>.................] - ETA: 12s - loss: 0.3203 - acc: 0.9043
28224/60000 [=============>................] - ETA: 12s - loss: 0.3192 - acc: 0.9047
28512/60000 [=============>................] - ETA: 12s - loss: 0.3179 - acc: 0.9051
28768/60000 [=============>................] - ETA: 12s - loss: 0.3175 - acc: 0.9051
28960/60000 [=============>................] - ETA: 12s - loss: 0.3167 - acc: 0.9054
29216/60000 [=============>................] - ETA: 12s - loss: 0.3160 - acc: 0.9056
29472/60000 [=============>................] - ETA: 11s - loss: 0.3156 - acc: 0.9057
29664/60000 [=============>................] - ETA: 11s - loss: 0.3149 - acc: 0.9059
29920/60000 [=============>................] - ETA: 11s - loss: 0.3138 - acc: 0.9064
30176/60000 [==============>...............] - ETA: 11s - loss: 0.3125 - acc: 0.9067
30496/60000 [==============>...............] - ETA: 11s - loss: 0.3114 - acc: 0.9071
30720/60000 [==============>...............] - ETA: 11s - loss: 0.3113 - acc: 0.9072
30976/60000 [==============>...............] - ETA: 11s - loss: 0.3106 - acc: 0.9073
31232/60000 [==============>...............] - ETA: 10s - loss: 0.3103 - acc: 0.9075
31488/60000 [==============>...............] - ETA: 10s - loss: 0.3093 - acc: 0.9078
31712/60000 [==============>...............] - ETA: 10s - loss: 0.3083 - acc: 0.9081
31968/60000 [==============>...............] - ETA: 10s - loss: 0.3075 - acc: 0.9083
32224/60000 [===============>..............] - ETA: 10s - loss: 0.3067 - acc: 0.9085
32448/60000 [===============>..............] - ETA: 10s - loss: 0.3059 - acc: 0.9086
32672/60000 [===============>..............] - ETA: 10s - loss: 0.3053 - acc: 0.9087
32896/60000 [===============>..............] - ETA: 10s - loss: 0.3041 - acc: 0.9091
33152/60000 [===============>..............] - ETA: 9s - loss: 0.3039 - acc: 0.9091 
33408/60000 [===============>..............] - ETA: 9s - loss: 0.3029 - acc: 0.9095
33600/60000 [===============>..............] - ETA: 9s - loss: 0.3028 - acc: 0.9096
33792/60000 [===============>..............] - ETA: 9s - loss: 0.3021 - acc: 0.9098
34080/60000 [================>.............] - ETA: 9s - loss: 0.3011 - acc: 0.9099
34304/60000 [================>.............] - ETA: 9s - loss: 0.3003 - acc: 0.9101
34592/60000 [================>.............] - ETA: 9s - loss: 0.2996 - acc: 0.9104
34816/60000 [================>.............] - ETA: 9s - loss: 0.2990 - acc: 0.9106
35104/60000 [================>.............] - ETA: 8s - loss: 0.2987 - acc: 0.9107
35424/60000 [================>.............] - ETA: 8s - loss: 0.2982 - acc: 0.9108
35648/60000 [================>.............] - ETA: 8s - loss: 0
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值