使用Python库进行多元多元回归

Question

在Python中是否有执行多元多元回归(具有多个因变量的多元回归)的库？

提前问候和感谢

Answer 1

你可以尝试sklearn中的模块，响应变量可以是2维或更多维的，我认为它适用于OLS (线性回归)，套索，岭。statsmodels中的模型只能做1个响应(刚刚检查)。

示例数据集：

import pandas as pd
from sklearn.datasets import load_iris
iris = load_iris()
df = pd.DataFrame(data= iris['data'],
                     columns= iris['feature_names'] )

df.shape
(150, 4)

现在我们做拟合：

from sklearn import linear_model
clf = linear_model.LinearRegression()
clf.fit(df[['sepal length (cm)']],df[['petal length (cm)','petal width (cm)']])
clf.coef_

array([[1.85843298],
       [0.75291757]])

在这种情况下，您可以看到系数与拟合一个响应时的系数相同：

clf.fit(df[['sepal length (cm)']],df[['petal width (cm)']])
clf.coef_
array([[0.75291757]])