import numpy as np
import sklearn.linear_model as lm
import sklearn.metrics as sm
import matplotlib.pyplot as mp
X,Y = [],[]
with open('dd.txt','r') as f:
for line in f.readlines():
data = [float(substr)
for substr in line.split(',')]
x.append(data[:-1])
y.append(data[-1])
x = np.array(x)
y = np.array(y)
#创建线性回归器
model = lm.LinearRegression()
#训练线性回归其
model.fit(x,y)
#预测输出
pred_y = model.predict(x)
mp.figure(num='Linear Regerssion',facecolor='lightgray')
mp.title('Linear Regerssion',fontsize = 20)
mp.xtable('x',fontsize=14)
mp.ylable('y',fontsize = 14)
mp.tick_params(labelsize=10)
mp.grid(linestyle=':')
mp.scatter(x,y,c='dodgerblue',s = 60,label='Sample')
mp.plot(x,pred_y,c='orangered',label='Prediction')
mp.legend()
mp.show()
ML3: sklearn 线性回归
最新推荐文章于 2021-02-27 14:21:52 发布