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