📌  相关文章
📜  AttributeError:模块 'tensorflow._api.v2.train' 没有属性 'GradientDescentOptimizer' - Python (1)

📅  最后修改于: 2023-12-03 14:39:23.631000             🧑  作者: Mango

AttributeError: module 'tensorflow._api.v2.train' has no attribute 'GradientDescentOptimizer' - Python

这个错误表示在使用 TensorFlow 时没有找到名为 GradientDescentOptimizer 的类或函数。在 TensorFlow 2.0 及更高版本中,优化器被移动到 tf.keras.optimizers 中。

改正这个错误的方法是将 from tensorflow._api.v2.train import GradientDescentOptimizer 更改为 from tensorflow.keras.optimizers import GradientDescent

以下是一些示例代码,可能会导致此错误:

import tensorflow as tf

...

optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.1)

要解决此错误,请将其更改为以下代码:

import tensorflow as tf

...

optimizer = tf.keras.optimizers.SGD(learning_rate=0.1)

还可以使用其他优化器,例如 AdamAdagrad 等。

import tensorflow as tf

...

optimizer = tf.keras.optimizers.Adam(learning_rate=0.001)

总之,在 TensorFlow 2.0 及更高版本中,必须使用 tf.keras.optimizers 和相应的优化器类。