线性回归示例代码,本示例预测身高与体重的关系。
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]]))