multiple-regression - scikit_learn 回归总结

Question

如果我在 R 中执行线性回归，我会得到一个很好的结果模型摘要、$R^2$、不同特征的 p 值等。

如果我在 scikit_learn 中做同样的事情，我将一无所获。有什么方法可以在那里打印模型摘要吗？

Answer 1

据我所知，Scikit-learn 没有像 R 那样的汇总功能。但是，另一个 Python 包 statmodels 有。另外，它的实现与 R 更相似。

from statsmodels.formula.api  import ols
#you need a Pandas dataframe df with columns labeled Y, X, & X2
est = ols(formula = 'Y ~  X + X2', data = df).fit()
est.summary()