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

Posted 2023-02-23

【中文标题】如何显示 LinearRegression() 的权重和偏差？

【英文标题】：How can I display the weights and bias from LinearRegression()?

【发布时间】：2019-09-26 02:36:18

【问题描述】：

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

【参考方案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