如果我在 R 中执行线性回归,我会得到一个很好的结果模型摘要、$R^2$、不同特征的 p 值等。
如果我在 scikit_learn 中做同样的事情,我将一无所获。有什么方法可以在那里打印模型摘要吗?
请您参考如下方法:
据我所知,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()
