python - 如何显示 LinearRegression() 的权重和偏差？

Question

我正在尝试解决线性回归问题，并且正在使用 sklearn 中的 LinearRegression() 函数。是否可以显示权重和偏差？

Answer 1

拟合模型后，使用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