# -*- coding: utf-8 -*-
from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt
# 载入数据
data = np.genfromtxt('一元线性回归.csv', delimiter=',')
x_data = data[:,0]
y_data = data[:,1]
plt.scatter(x_data,y_data)
plt.show()
#x_data = data[:,0:1]
#x_data = data[:,0,np.newaxis]
x_data
x_data = data[:,0,np.newaxis]
y_data = data[:,1,np.newaxis]
# 创建并拟合模型
model = LinearRegression()
model.fit(x_data, y_data)
coef = model.coef_ #获得该回该方程的回归系数与截距
intercept = model.intercept_
print("预测方程回归系数:", coef)
print("预测方程截距:", intercept)
# 画图
plt.plot(x_data, y_data, 'g.')
plt.plot(x_data, model.predict(x_data), 'r')
plt.show()
文章来源地址https://www.toymoban.com/news/detail-821387.html
文章来源:https://www.toymoban.com/news/detail-821387.html
到了这里,关于python一元线性回归sklearn的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!