使用GradientDescentOptimizer而不是Adam Optimizer时,模型似乎不会收敛.另一方面,AdamOptimizer似乎工作正常.来自tensorflow的GradientDescentOptimizer出了什么问题?
import matplotlib.pyplot as plt
import tensorflow as tf
import numpy as np
def randomSample(size=100):
"""
y = 2 * x -3
"""
x = np.random.randint(500, size=size)
y = x * 2 - 3 - np.random.randint(-20, 20, size=size)
return x, y
def plotAll(_x, _y, w, b):
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(_x, _y)
x = np.random.randint(500, size=20)
y = w * x + b
ax.plot(x, y,'r')
plt.show()
def lr(_x, _y):
w = tf.Variable(2, dtype=tf.float32)
b = tf.Variable(3, dtype=tf.float32) …Run Code Online (Sandbox Code Playgroud) python regression machine-learning gradient-descent tensorflow