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