线性回归示例代码

线性回归示例代码,本示例预测身高与体重的关系。

from sklearn.linear_model import LinearRegression

# 1、准备数据,案例是关于身高与体重的预测。x是身高特征,y是体重目标
x = [[160],[166],[172],[174],[180]]
y = [56.3,60.6,65.1,68.5,75]

# 2、创建线性回归模型
model = LinearRegression()

# 3、将模型训练
model.fit(x,y)

# coef_斜率(权重),intercept_截距(偏置)
print(model.coef_)
print(model.intercept_)

# 4、模型预测,传入身高特征
print(model.predict([[176]]))