TensorFlow 是一个开源的机器学习框架,由 Google 开发。它可以帮助开发者轻松构建和训练复杂的机器学习模型。以下是 TensorFlow 入门的简要教程。
安装 TensorFlow
在开始之前,您需要先安装 TensorFlow。您可以通过以下命令进行安装:
pip install tensorflow
简单示例
以下是一个使用 TensorFlow 创建简单线性回归模型的示例:
import tensorflow as tf
# 创建 TensorFlow 图
x = tf.constant([1, 2, 3, 4])
y = tf.constant([1, 2, 3, 4])
# 创建线性模型
model = tf.keras.Sequential([tf.keras.layers.Dense(units=1, input_shape=[1])])
# 编译模型
model.compile(optimizer='sgd', loss='mean_squared_error')
# 训练模型
model.fit(x, y, epochs=1000)
# 预测
print(model.predict([5]))
学习资源
如果您想了解更多关于 TensorFlow 的信息,可以访问以下链接:
TensorFlow Logo
扩展阅读
希望这个入门教程对您有所帮助!