始める無料で始める

線形回帰モデルの構築

特徴量と目的変数の配列を作成したので、次はすべての特徴量と目的変数の値を使って線形回帰モデルを学習させましょう。

ここでの目的は特徴量と目的変数の関係を調べることなので、データを訓練セットとテストセットに分割する必要はありません。

Xy は以下のようにあらかじめ読み込まれています:

y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)

この演習はコースの一部です

scikit-learn による教師あり学習

コースを見る

演習の手順

  • LinearRegression をインポートしましょう。
  • 線形回帰モデルをインスタンス化しましょう。
  • X を使って売上値を予測し、結果を predictions に格納しましょう。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Import LinearRegression
from ____.____ import ____

# Create the model
reg = ____()

# Fit the model to the data
____

# Make predictions
predictions = ____

print(predictions[:5])
コードを編集して実行