keras实现线性回归

env

windows+anaconda+jupyter
tensorflow2.3+keras

#导入库
import keras
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from keras import layers
%matplotlib inline

#准备数据
x = np.linspace(0,100,30)   #在指定的间隔内返回均匀间隔的数字。
y = 3*x+7+np.random.randn(30)

#建立顺序模型
model = keras.Sequential()
#加入线性层
model.add(layers.Dense(1,input_dim=1))

#编译模型
model.compile(optimizer = 'adam',
             loss = 'mse',
             )

#训练模型
model.fit(x,y,epochs = 5000)

y2 = model.predict(x)
plt.scatter(x,y)
plt.plot(x,y2)