TensorFlow 是一个开源的机器学习框架,由 Google Brain 团队开发。它广泛应用于各种机器学习任务,包括深度学习。本教程将带你了解 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], [1, 2, 3, 4], epochs=100)
# 评估模型
print(model.evaluate([5], [5]))
更多资源
想要了解更多关于 TensorFlow 的内容,可以访问我们的 TensorFlow 教程 页面。
图片示例
下面是一个 TensorFlow 的图标,它可以帮助你更好地理解 TensorFlow。