Pythonで線形回帰を実装する

線形回帰の概念を理解したところで、Pythonを使って実装してみましょう。ここでは、この目的に広く使われているscikit-learnライブラリを使用します。

ステップ1：データを準備する

まず、データセットを準備する必要があります。この例では、1つの独立変数（x）と1つの従属変数（y）を持つ単純なデータセットを使用します。

データは次のとおりです。

• x：1, 2, 3, 4, 5
• y：2, 4, 6, 8, 10

ステップ2：必要なライブラリをインポートする

次のライブラリをインポートする必要があります。

• numpy：数値計算に使用します。
• matplotlib：可視化に使用します。
• scikit-learn：機械学習アルゴリズムに使用します。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

ステップ3：モデルを作成する

scikit-learnのLinearRegressionクラスを使って線形回帰モデルを作成し、データに適合させます。

x = np.array([1, 2, 3, 4, 5]).reshape(-1, 1)
y = np.array([2, 4, 6, 8, 10])

model = LinearRegression()
model.fit(x, y)

ステップ4：予測する

モデルの学習が完了したら、それを使って予測できます。たとえば、x = 6のときのyの値を予測します。

prediction = model.predict(np.array([[6]]))
print(f"Predicted value for x=6: {prediction[0]}")

ステップ5：結果を可視化する

データ点と回帰直線をプロットして、モデルがデータにどの程度適合しているかを可視化できます。

plt.scatter(x, y, color='blue', label='Data points')
plt.plot(x, model.predict(x), color='red', label='Regression line')
plt.title('Linear Regression Visualization')
plt.xlabel('X')
plt.ylabel('Y')
plt.legend()
plt.show()

ステップ6：モデルを評価する

決定係数（R-squared）などの指標を使って、モデルの性能を評価します。決定係数は、モデルがデータの変動をどの程度説明できているかを測定します。

r_squared = model.score(x, y)
print(f"R-squared value: {r_squared}")

実装における一般的な課題

線形回帰を実装する際には、次のような課題に直面することがあります。

• 小規模なデータセットでは過学習が起こりやすくなります。
• 外れ値の影響を受けやすくなります。
• 複数の変数における多重共線性への対処が必要になります。

まとめと次のステップ

このレッスンでは、次の内容を学びました。

• scikit-learnを使用して線形回帰を実装しました。
• 予測を行い、結果を可視化しました。
• 決定係数（R二乗）を使用してモデルを評価しました。

次は、分類問題に使用するロジスティック回帰の概念について学びます。