TensorFlow 是一个由 Google 开发的开源机器学习框架,用于构建和训练各种机器学习模型。本教程将为您介绍 TensorFlow 的基础知识。
快速开始
安装 TensorFlow
在开始之前,您需要安装 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.predict([6]))
深入学习
要深入了解 TensorFlow,您可以访问我们的 TensorFlow 教程。
TensorFlow Logo
如果您有任何关于 TensorFlow 的问题,欢迎在 社区论坛 中提问。