TensorFlow 是一个开源的机器学习框架,由 Google 的大脑团队开发。GitHub 上的 TensorFlow 文档提供了丰富的教程、API 文档和社区资源。

安装 TensorFlow

在安装 TensorFlow 之前,请确保您的系统满足以下要求:

  • Python 3.6 或更高版本
  • 确保您的 Python 环境是干净的,没有其他版本的 TensorFlow

以下是一个简单的安装命令:

pip install tensorflow

示例代码

这是一个简单的 TensorFlow 示例,用于创建一个线性回归模型:

import tensorflow as tf

# 创建一个线性回归模型
model = tf.keras.Sequential([
    tf.keras.layers.Dense(units=1, input_shape=[1])
])

# 编译模型
model.compile(optimizer='sgd', loss='mean_squared_error')

# 训练模型
model.fit([1, 2, 3, 4, 5], [1, 2, 3, 4, 5], epochs=100)

# 评估模型
print(model.evaluate([1, 2, 3, 4, 5], [1, 2, 3, 4, 5]))

获取更多帮助

如果您需要更多帮助,请访问我们的 TensorFlow 官方文档

TensorFlow Logo