AI 上級

75. scikit-learnでモデルを学習・予測する

部屋の広さから家賃を予測してみる

前のレッスンで学んだ流れを、実際にscikit-learnのコードで体験してみましょう。今回は、部屋の広さから家賃を予測する、というシンプルな例で流れをつかみます。

サンプルコード

from sklearn.linear_model import LinearRegression

# 部屋の広さ（m^2）
X = [[20], [30], [40], [50]]
# 家賃（万円）
y = [5, 7, 9, 11]

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

# 35m^2の部屋の家賃を予測
print(model.predict([[35]]))
LinearRegression（線形回帰）という、最もシンプルな機械学習モデルを使っています。fit(X, y)で「広さと家賃の関係」を学習させ、predict()で学習していない新しい広さ（35m^2）に対する家賃を予測しています。

📝 練習問題

練習問題：X = [[1], [2], [3]]、y = [100, 200, 300]というデータでLinearRegressionモデルを学習させ、4に対する予測値を表示してください。