梯度下降法 - sklearn一元线性回归

代码是自己在学习网课时跟着敲的，数据源data.csv文件我会放在我的资源里，有兴趣的可以自己下载试试

from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt
#载入数据,delimite表示分隔符为逗号
data=np.genfromtxt("data.csv",delimiter=",")
x_data=data[:,0]
y_data=data[:,1]
plt.scatter(x_data,y_data)
plt.show()
print(x_data.shape)
x_data=data[:,0,np.newaxis]
y_data=data[:,1,np.newaxis]
#创建并拟合模型
model=LinearRegression()
model.fit(x_data,y_data) 
#画图
plt.plot(x_data,y_data,'b.')
plt.plot(x_data,model.predict(x_data),'r')
plt.show()

最终运行结果如下：
注：如有侵权，请联系删除！