📅  最后修改于: 2023-12-03 14:39:23.631000             🧑  作者: Mango
这个错误表示在使用 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)
还可以使用其他优化器,例如 Adam
,Adagrad
等。
import tensorflow as tf
...
optimizer = tf.keras.optimizers.Adam(learning_rate=0.001)
总之,在 TensorFlow 2.0 及更高版本中,必须使用 tf.keras.optimizers
和相应的优化器类。