scikit_learn中的回归总结

时间:2015-10-14 22:41:31

标签: multiple-regression python scikit-learn

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

如果我在scikit_learn中做同样的事情,我什么也得不到。有没有办法在那里打印模型摘要?

1 个答案:

答案 0 :(得分:5)

据我所知,Scikit-learn没有像R这样的汇总函数。但是,statmodels是另一个Python软件包。此外,它的实现与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()