【发布时间】:2020-06-22 13:37:28
【问题描述】:
我正在使用一个基本的 NN 来训练和测试 MNIST 数据集的准确性。
系统:i5 第 8 代,GPU - Nvidia 1050Ti
这是我的代码:
from __future__ import print_function,absolute_import,unicode_literals,division
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train,y_train) , (x_test,y_test) = mnist.load_data()
#x_train , y_train = x_train/255.0 , y_train/255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28,28)),
tf.keras.layers.Dense(312,activation='relu'),
tf.keras.layers.Dropout(0.3),
tf.keras.layers.Dense(10,activation='softmax')
])
model.compile(
optimizer='Adamax',
loss="sparse_categorical_crossentropy",
metrics=['accuracy']
)
model.fit(x_train,y_train,epochs=5)
model.evaluate(x_test,y_test)
当我像第 5 行那样对图像进行归一化时,准确度会急剧下降:
loss: 10392.0626 - accuracy: 0.0980
但是,当我不规范化它们时,它会给出:
- loss: 0.2409 - accuracy: 0.9420
一般来说,归一化数据有助于梯度下降更快地收敛。为什么会有这么大的差异? 我错过了什么?
【问题讨论】:
-
为什么要将标签除以 255?
-
正如GoodDeeds所说,您不能将标签除以255。标签是0、1、2等。那是错误的。此外,您忘记标准化 x_test。
标签: python tensorflow deep-learning neural-network mnist