Laravel  
laravel
文档
数据库
架构
入门
php技术
    
Laravelphp
laravel / php / java / vue / mysql / linux / python / javascript / html / css / c++ / c#

tensorflow python

作者:索伦之刃   发布日期:2025-07-25   浏览:16

# 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}')

解释说明:

  1. 加载数据:我们使用 tf.keras.datasets.mnist 来加载 MNIST 手写数字数据集,并将其分为训练集和测试集。
  2. 数据预处理:将图像数据从 [0, 255] 归一化到 [0, 1] 的范围,以加快训练速度和提高模型性能。
  3. 构建模型:使用 Sequential 模型,依次添加 Flatten 层(展平输入)、Dense 层(全连接层)、Dropout 层(防止过拟合)和输出层(分类层)。
  4. 编译模型:选择优化器、损失函数和评估指标。这里使用了 Adam 优化器和稀疏分类交叉熵损失函数。
  5. 训练模型:使用 fit 方法训练模型,指定训练数据、标签以及训练轮数(epochs)。
  6. 评估模型:使用测试集评估模型的性能,并输出测试准确率。

如果你需要更多关于 TensorFlow 的信息或更复杂的示例,请参考官方文档或相关教程。

上一篇:python图

下一篇:python pyserial

大家都在看

python时间格式

python开发windows应用程序

python中len是什么意思

python ord和chr

python中的yield

python自定义异常

python判断路径是否存在

python list.pop

python的for i in range

npm config set python

Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3

Laravel 中文站