前提・実現したいこと
Python、統計学初心者の者です。
仕事の関係で、下記のことをPythonで実現したいと考えております。
① あるデータをtrainデータとtestデータに分割する
② trainデータを用いて、GLM(一般化線形モデル)によるポアソン回帰モデルを作成する。
③ 上記のモデルにtestデータを代入し、testデータの実測値とモデルによる予測値を比較する
(交差検証:cross variation)
今回、testデータを1つのみとし、trainデータを残りのすべてとする手法(leave-one-out)を施行したいと考えております。
発生している問題・エラーメッセージ
まず、お弁当の需要予測(https://signate.jp/competitions/24)のデータを用いて、
上記の手法をトライしました。
Python初心者で思い通りのコードを書くことはできないので、
インターネットからコードのコピペを組み合わせて試行錯誤しました。
(https://tanuhack.com/statsmodels-multiple-lra/ を参考にしました。)
まず、0行目のデータ(2013/11/18のデータ)をtestデータ、残りをtrainデータとしてコードを書きました。
(このコードがうまく動けば、i行目のiをfor構文で回して、すべての行に対してこれを行い、評価する予定です。)
気温を説明変数、お弁当の販売数を応答変数とするポアソン回帰モデルを作成しました。
すると、以下のようなエラーメッセージが出現します。
ValueError Traceback (most recent call last) <ipython-input-5-d4fb5ef762d3> in <module> 29 testX = sm.add_constant(testX) 30 ---> 31 pred = result.predict(testX) 32 test['pred'] = pred 33 test ~\anaconda3\lib\site-packages\statsmodels\base\model.py in predict(self, exog, transform, *args, **kwargs) 1098 1099 predict_results = self.model.predict(self.params, exog, *args, -> 1100 **kwargs) 1101 1102 if exog_index is not None and not hasattr(predict_results, ~\anaconda3\lib\site-packages\statsmodels\genmod\generalized_linear_model.py in predict(self, params, exog, exposure, offset, linear) 870 exog = self.exog 871 --> 872 linpred = np.dot(exog, params) + offset + exposure 873 if linear: 874 return linpred <__array_function__ internals> in dot(*args, **kwargs) ValueError: shapes (1,1) and (2,) not aligned: 1 (dim 1) != 2 (dim 0)
該当のソースコード
Python
1# 1. 必要なライブラリの読み込み 2import pandas as pd 3import statsmodels.formula.api as smf 4import statsmodels.api as sm 5from matplotlib import pyplot as plt 6 7i = 0 8 9# 2. 使用するデータの読み込み 10df = pd.read_csv("otehon data/trainmoji.csv", encoding="shift-jis") 11 12test = df.iloc[[i]] #i行目をtestデータとする 13train = df.drop(test.index) #残りをtrainデータとする 14 15# 3. smf.glmで使うformula(線形予測子)とfamily(確率分布)を設定する 16trainY = train['y'] # 目的変数:y 17 18trainX = pd.get_dummies(train[['temperature']]) # 説明変数 19family = sm.families.Poisson() 20trainX = sm.add_constant(trainX) 21 22# 4. 先ほどの設定値を使って一般化線形モデルを作成 23model = sm.GLM(trainY, trainX, family=family) 24 25# 5. 作成したモデルを学習させる 26result = model.fit() 27 28# 6. testデータで検証 29testX = pd.get_dummies(test[['temperature']]) # 説明変数 30testX = sm.add_constant(testX) 31 32pred = result.predict(testX) 33test['pred'] = pred 34test
試したこと
いろいろ調べると、配列の要素数があっていないときにこのようなメッセージが出るようですが、
train, testに格納された列数は一致しているはずです。
補足情報(FW/ツールのバージョンなど)
ここにより詳細な情報を記載してください。
回答1件
あなたの回答
tips
プレビュー