我正在尝试解决线性回归问题,并且正在使用LinearRegression()
中的sklearn
函数。可以显示权重和偏差吗?
答案 0 :(得分:1)
一旦fit
,该模型就会使用coef_
属性检索权重,并使用intercept_
获得偏差项。
请参见以下示例:
import numpy as np
from sklearn.linear_model import LinearRegression
a = np.array([[5,8],[12,24],[19,11],[10,15]])
## weights
w = np.array([0.2, 0.5])
## bias
b = 0.1
y = np.matmul(w, a.T) + b
lr = LinearRegression()
lr.fit(a, y)
print(lr.coef_)
# array([0.2, 0.5])
print(lr.intercept_)
# 0.099
有关更多详细信息,请参见documentation