sklearnを使って単回帰分析をしようと思ったらエラーが出る

Question

以下のコードを実行したら
ValueError: matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 5 is different from 1)
と出力されます。これは何に問題があると考えられるのでしょうか？

import numpy as np
from sklearn.linear_model import LinearRegression

lr = LinearRegression()
x = np.array([
              [
               3,5,7,9,10
               ]
              ])
t = np.array([
              [1,3,5,7,9]
              ])

lr.fit(x,t)

x = np.array([
              [4]
              ])
lr.predict(x)

Answer 1

こういうことかな?

import numpy as np
from sklearn.linear_model import LinearRegression

lr = LinearRegression()
x = np.array([
              [
               3,5,7,9,10
               ]
              ])
t = np.array([
              [1,3,5,7,9]
              ])

lr.fit(x.T,t.T)

x2 = np.array([
              [4]
              ])
lr.predict(x2)
# array([[1.92682927]])

Comments

[jbpb0]

lr.coef_ * 4 + lr.intercept_
と同じ結果になります

[Moai]

そのような書き方があったんですね！
fit関数のところで両方とも転置をしているのはどうしてでしょうか？

[jbpb0]

x, tは、縦に数値が並ばないといけないからです
お書きになったやり方だと横に数値が並んでいるので、転置して縦並びにしました
下記のようにして、サイズを確認してみてください
print(x.shape)
print(t.shape)

また、下記のようにすれば、縦並びなので転置は不要です
x = np.array([ [3],[5],[7],[9],[10] ])
t = np.array([ [1],[3],[5],[7],[9] ])
lr.fit(x,t)
x2 = np.array([ [4] ])
lr.predict(x2)

https://arakan-pgm-ai.hatenablog.com/entry/2019/07/12/000000
の真ん中あたりの「学習と評価でエラーがでる場合があります」のところも見てください

Answer 2

x = np.array([[4]])のところで要素数を5にしてください。(例えばx = np.array([[4,2,9,1,7]]))

Comments

[Moai]

ご返信ありがとうございます。
単回帰分析をしたい場合はlrのcoef_とintercept_を求めた後、それを元に式を作り、引数を算出するということでした。
つまり、
y =lr.corf*4_ + intercept_
で求められるということでした。