# TensorFlow 是一个用于机器学习和深度学习的开源库。
# 下面是一个简单的 TensorFlow Python 示例代码,用于创建一个基本的神经网络模型来对 MNIST 数据集进行分类。
import tensorflow as tf
from tensorflow.keras import layers, models
# 加载 MNIST 数据集
mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 数据预处理:将图像数据归一化到 [0, 1] 范围
x_train, x_test = x_train / 255.0, x_test / 255.0
# 构建神经网络模型
model = models.Sequential([
layers.Flatten(input_shape=(28, 28)), # 将输入的 28x28 图像展平为 784 维向量
layers.Dense(128, activation='relu'), # 全连接层,使用 ReLU 激活函数
layers.Dropout(0.2), # Dropout 层,防止过拟合
layers.Dense(10, activation='softmax') # 输出层,使用 softmax 激活函数进行多分类
])
# 编译模型
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
# 训练模型
model.fit(x_train, y_train, epochs=5)
# 评估模型
test_loss, test_acc = model.evaluate(x_test, y_test, verbose=2)
print(f'\nTest accuracy: {test_acc}')
tf.keras.datasets.mnist
来加载 MNIST 手写数字数据集,并将其分为训练集和测试集。Sequential
模型,依次添加 Flatten 层(展平输入)、Dense 层(全连接层)、Dropout 层(防止过拟合)和输出层(分类层)。fit
方法训练模型,指定训练数据、标签以及训练轮数(epochs)。如果你需要更多关于 TensorFlow 的信息或更复杂的示例,请参考官方文档或相关教程。
上一篇:python图
下一篇:python pyserial
Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3
Laravel 中文站