我在尝试解决一个线性回归问题,并且使用了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
更多详情请参考文档