如何显示LinearRegression()的权重和偏差?

时间:2019-05-08 11:10:44

标签: python machine-learning scikit-learn linear-regression

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

1 个答案:

答案 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