用python编写数字识别
时间: 2023-07-16 21:15:17 浏览: 197
可以使用Python中的机器学习库TensorFlow来编写数字识别程序。以下是一个简单的示例代码:
```python
import tensorflow as tf
from tensorflow import keras
# 加载MNIST数据集
mnist = keras.datasets.mnist
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
# 对图像进行归一化处理
train_images = train_images/255.0
test_images = test_images/255.0
# 定义模型
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28,28)),
keras.layers.Dense(128, activation=tf.nn.relu),
keras.layers.Dense(10, activation=tf.nn.softmax)
])
# 编译模型
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
# 训练模型
model.fit(train_images, train_labels, epochs=5)
# 评估模型
test_loss, test_acc = model.evaluate(test_images, test_labels)
print('Test accuracy:', test_acc)
# 使用模型进行预测
predictions = model.predict(test_images)
```
该程序使用MNIST数据集进行训练和测试,该数据集包含手写数字图像及其相应的标签。首先将图像像素值归一化为0到1之间的值,然后定义一个包含两个密集层的神经网络模型。使用adam优化器和稀疏分类交叉熵损失函数对模型进行编译。接下来,在训练集上训练模型,并在测试集上评估模型的性能。最后,使用模型进行预测。
阅读全文