LinearRegressionを使って回帰分析を行いたいです。入力データについて教えてください。

Question

前提・実現したいこと

単純なデータを用いて、LinearRegressionの動きを確認しようとしているのですが、
以下のエラーメッセージが発生しました。
単純なミスなきもしているのですが、自分ではわからないため、どこがおかしいのか教えて頂ければと考えています。
宜しくお願いします。

発生している問題・エラーメッセージ

ValueError: Expected 2D array, got 1D array instead:
array=[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14].
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.

該当のソースコード

import numpy as np
from sklearn.linear_model import LinearRegression

a = np.arange(15)
b = np.arange(15)

slr = LinearRegression()
slr.fit(a, b)
y = slr.predict(16)

print(y)

試したこと

エラーメッセージがexpected 2dとなっているため、
a = np.arange(30).reshape(15,2)の２次元にしてもうまくいきません

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

エラーメッセージで示されている通り、fitに渡す引数は(要素の数,1)の形にする必要があります。
具体的にはa = a.reshape(-1,1)とすればよいです。bも同様です。
また、predictの引数も同様ですので、y = slr.predict(np.array([16]).reshape(-1,1))とする必要があります。

Comments

[yamahisa]

動かすことができました。
また、エラーメッセージを読もうという気があれば解決できてた問題だということも理解できました。

丁寧な回答ありがとうございました！