时间戳

首页 > 解决方案 > Scikit 学习多元线性回归和多项式特征的系数顺序

python - Scikit 学习多元线性回归和多项式特征的系数顺序

问题描述

我正在拟合一个简单的多项式回归模型，我想从拟合模型中获取系数。

给定准备代码：

import pandas as pd
from itertools import product
from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
from sklearn.pipeline import make_pipeline

# data creation
sa = [1, 0, 1, 2, 3]
sb = [2, 1, 0, 1, 2]
raw = {'a': [], 'b': [], 'w': []}
for (ai, av), (bi, bv) in product(enumerate(sa), enumerate(sb)):
    raw['a'].append(ai)
    raw['b'].append(bi)
    raw['w'].append(av + bv)
data = pd.DataFrame(raw)

# regression
x = data[['a', 'b']].values
y = data['w']
poly = PolynomialFeatures(2)
linr = LinearRegression()
model = make_pipeline(poly, linr)
model.fit(x, y)

从这个答案中，我知道可以使用 with 获得系数

model.steps[1][1].coef_
>>> array([  0.00000000e+00,  -5.42857143e-01,  -1.71428571e+00,
             2.85714286e-01,   1.72774835e-16,   4.28571429e-01])

但这提供了一个一维数组，我不确定哪些数字对应于哪些变量。

它们是按a 0、 a 1、 a 2、 b 0、 b 1、 b 2还是按a 0、 b 0、 a 1、 b 1、 a 2、 b 2排序的？

标签： pythonscikit-learnregression

解决方案

您可以使用 的get_feature_names()来PolynomialFeatures了解订单。

在管道中，您可以这样做：

model.steps[0][1].get_feature_names()

# Output:
['1', 'x0', 'x1', 'x0^2', 'x0 x1', 'x1^2']

如果您有功能的名称（在您的情况下为“a”、“b”），您可以通过它来获取实际功能。

model.steps[0][1].get_feature_names(['a', 'b'])

# Output:
['1', 'a', 'b', 'a^2', 'a b', 'b^2']

推荐阅读