如何在 python 中找到基础扩展矩阵

How to you find the basis expansion matrix in python

我正在进行非参数回归,需要一个函数来将设计矩阵 X 扩展到基础矩阵中。有没有可以做到这一点的软件包?

例如,如果 X 是 200*10(200 个观测值和 10 个特征),使用具有 5 个基的 B 样条基扩展将产生一个 200*50 基矩阵。

我试过scipy.interpolate.BSpline,但好像没有return基矩阵。

python 中的 pasty 库很有用。

from pasty import dmatrix

transformed_x = dmatrix(
                    "bs(x, df=df, degree=degree, include_intercept=False)",
                    {"train": x}, return_type='matrix')

这将 return 向量 x 的基础展开。如果您有数据矩阵,请对每一列执行此操作。