tensorflow2 實現簡單的線性迴歸

話不多說，直接上代碼

import tensorflow as tf
import numpy as np

input_x=np.random.rand(1000)
input_y=5*input_x+0.227
weight=tf.Variable(1.0,dtype=tf.float32,name="weight")
bias=tf.Variable(1.0,dtype=tf.float32,name="bias")

def model(x):
    y=tf.multiply(x,weight)+bias
    return y
opt=tf.optimizers.Adam(1e-1)

for xs,ys in zip(input_x,input_y):
    xs=np.reshape(xs,[1])
    ys=np.reshape(ys,[1])
    loss=lambda :tf.losses.MeanSquaredError()(model(xs),ys)
    opt.minimize(loss,[weight,bias])
    print(loss().numpy())

print(weight)
print(bias)