要素具有固定系数的多元线性回归

时间:2019-11-13 23:43:43

标签: python regression linear-regression coefficients

具有两个特征的线性回归可以通过以下等式描述:

y = a1x1 + a2x2 +拦截

拟合多元线性回归将求解系数a1a2。考虑以下代码:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn import linear_model

file = 'https://aegis4048.github.io/downloads/notebooks/sample_data/unconv_MV_v5.csv'
df = pd.read_csv(file)[['Por', 'Perm', 'Prod']]

features = df[['Por', 'Perm']].values.reshape(-1,2)
target = df['Prod']

ols = linear_model.LinearRegression()
model = ols.fit(features, target)
predicted = model.predict(features)

coef = model.coef_

pd.DataFrame(coef, index=['Por', 'Perm'], columns=['Regression Coef']).round(2)

>>>         Regression Coef
    Por              244.47
    Perm              97.75

这两个功能是PorPerm。我想将Perm的回归系数的值固定为某个固定值,并且仅求解Por的系数。如何在Python中执行此操作?

1 个答案:

答案 0 :(得分:2)

Pora2。将a2的值设置为固定值A2后,线性回归将减少为y(a1) = a1x1 + (A2x2 + intercept)。因此,您可以简单地解决简单的线性回归y(a1) = a1x1 + intercept_new,其中intercept_new已经考虑到将Por设置为恒定值。