1. ホーム
  2. python

[解決済み] statsmodelsのOLSフィットで、パラメータが1つしか得られないのはなぜですか?

2022-02-17 01:41:49

質問

以下は、私がやっていることです。

$ python
Python 2.7.6 (v2.7.6:3a1db0d2747e, Nov 10 2013, 00:42:54) 
[GCC 4.2.1 (Apple Inc. build 5666) (dot 3)] on darwin
>>> import statsmodels.api as sm
>>> statsmodels.__version__
'0.5.0'
>>> import numpy 
>>> y = numpy.array([1,2,3,4,5,6,7,8,9])
>>> X = numpy.array([1,1,2,2,3,3,4,4,5])
>>> res_ols = sm.OLS(y, X).fit()
>>> res_ols.params
array([ 1.82352941])

私は2つの要素を持つ配列を期待していたのですが...!? 切片とスロープ係数?

どのように解決するのですか?

これを試してみてください。

X = sm.add_constant(X)
sm.OLS(y,X)

のように ドキュメント :

インターセプトはデフォルトでは含まれないので、ユーザーが追加する必要があります。

statsmodels.tools.tools.add_constant