我在尝试解决一个线性回归问题,并且使用了sklearn
中的LinearRegression()
函数。是否可以显示权重和偏置?
回答:
一旦你使用fit
方法训练模型后,可以使用coef_
属性来获取权重,使用intercept_
来获取偏置项。
请看下面的示例:
import numpy as npfrom sklearn.linear_model import LinearRegression a = np.array([[5,8],[12,24],[19,11],[10,15]])## 权重w = np.array([0.2, 0.5])## 偏置b = 0.1 y = np.matmul(w, a.T) + blr = LinearRegression()lr.fit(a, y)print(lr.coef_)# array([0.2, 0.5])print(lr.intercept_)# 0.099
更多详情请参考文档