TensorFlow 是一个由 Google 开源的机器学习框架,广泛应用于各种机器学习和深度学习任务。以下是一个简单的 TensorFlow 教程,帮助您入门。
安装 TensorFlow
在开始之前,您需要安装 TensorFlow。您可以从 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, 2.5, 3, 3.5], epochs=100)
# 使用模型进行预测
print(model.predict([10]))
扩展阅读
如果您想深入了解 TensorFlow,可以阅读以下资源:
图片展示
TensorFlow 模型训练中的图像: