Skip to main content
 首页 » 编程设计

python之模块 'tensorflow._api.v2.train' 没有属性 'GradientDescentOptimizer'

2024年11月24日25mfryf

我用的是Python 3.7.3,安装了tensorflow 2.0.0-alpha0,但是有一些问题。比如 模块“tensorflow._api.v2.train”没有属性“GradientDescentOptimizer” 这是我所有的代码

import  tensorflow as tf 
import  numpy as np 
 
x_data=np.random.rand(1,10).astype(np.float32) 
y_data=x_data*0.1+0.3 
 
 
Weights = tf.Variable(tf.random.uniform([1], -1.0, 1.0)) 
biases = tf.Variable(tf.zeros([1])) 
y=Weights*x_data+biases 
 
loss=tf.reduce_mean(tf.square(y-y_data)) 
 
optimizer=tf.train.GradientDescentOptimizer(0.5) 
train=optimizer.minimize(loss) 
 
init = tf.global_variables_initializer()   
 
sess = tf.Session() 
sess.run(init)           
 
for step in range(201): 
    sess.run(train) 
    if step % 20 == 0: 
        print(step, sess.run(Weights), sess.run(biases)) 

请您参考如下方法:

在 TensorFlow 2.0 中,Keras 成为默认的高级 API,优化器函数从 tf.keras.optimizers 迁移到名为 tf.optimizers 的单独 API .它们继承自 Keras 类 Optimizer。 tf.train 中的相关函数未包含在 TF 2.0 中。因此,要访问 GradientDescentOptimizer,请调用 tf.optimizers.SGD