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
  • Neural Network