from sklearn
.linear_model
import LinearRegression
import pandas
as pd
import numpy
as np
import matplotlib
.pyplot
as plt
data
= pd
.read_excel
('./car_data.xls')
lr
= LinearRegression
()
X
= data
.loc
[:,'TV Ads']
y
= data
.loc
[:,'Cars Sold']
X
= (X
.values
).reshap
(-1,1)
lr
.fit
(X
,y
)
k
= lr
.coef_
[0]
print(k
)
b
= lr
.intercept_
print(b
)
y_predict
= lr
.predict
(X
)
print(y_predict
)
print(plt
.scatter
(X
,y
))
print(plt
.plot
(X
,y_predict
))
创建模型实例将数据进行特征工程的处理(1-D转换成2-D)将特征工程处理过后的数据,进行模型的训练利用生成的模型进行预测
转载请注明原文地址: https://lol.8miu.com/read-14004.html