【发布时间】:2021-08-10 23:03:58
【问题描述】:
MNIST 数据集包含所有 10 个要训练的数字。如果我预测9,模型将输出9。但是,如果我想预测数字34.542 怎么办?它会给我一个错误的数字,因为我只训练了 0 到 9。那么,我如何预测 > 9 数字?
这是我的代码,但我认为它在这里没有用处
model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(units=128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(units=128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(units=10, activation=tf.nn.softmax))
# Compiling and optimizing model
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
# Training the model
model.fit(X_train, y_train, epochs=3)
【问题讨论】:
-
你不能期望得到一个训练集中不存在的数字的正确预测。
-
但是
34.542的每个数字都可以分段吗?
标签: python tensorflow keras mnist