最佳拟合功能无法正确绘制

Question

我修改了线性回归代码，并将预测函数从线性更改为二次。但是，当我尝试绘制最佳拟合线时，它无法正确绘制。

pred = tf.add(tf.add(tf.multiply(tf.pow(X, 2), W1), tf.multiply(X, W2)), b)

• 我将其从y = Wx + b函数更改为y = W1X ^ 2 + W2X + b。我添加了W1和X ^ 2。

print("Epoch:" + str((epoch+1)), "cost=" + str(c),  "W1=" + str(sess.run(W1)),"W2=" + str(sess.run(W2)), "b=" + str(sess.run(b)))

• 我在这里添加了新创建的体重

    plt.plot(trainX, (sess.run(W1)*(trainX**2)) + sess.run(W2)*trainX + sess.run(b), label='Fitted line')

• 这是我画线的地方，我很确定错误在这里。 TrainX只是包含我的训练点的所有X值的列表。

https://i.stack.imgur.com/EE22B.png

• 这是我最合适的系列。它看起来不像普通的二次函数