1. 学ぶ
  2. /
  3. コース
  4. /
  5. Pythonで学ぶマーケティングのための機械学習

Connected

演習

データを学習用とテスト用に分割する

回帰モデルの構築に進む前の最終ステップです!ここでは、目的変数と特徴量列の名前を特定し、データを抽出して、学習用とテスト用に分割します。

pandas と numpy ライブラリはそれぞれ pd、np として読み込まれています。入力特徴量は features データセットとしてインポートされており、前の演習で作成した目的変数は Y として用意されています。

指示

100 XP
  • 顧客IDの列名をリストとして保存します。
  • 顧客IDを除外して、特徴量の列名を選択します。
  • 特徴量を X として抽出します。
  • train_test_split() 関数を使って、データを学習用とテスト用に分割します。