1. 学ぶ
    2. /
  2. コース
    3. /
  3. scikit-learn で学ぶ Supervised Learning
Connected

演習

訓練/テスト分割と精度の計算

churn_df データセットを使って、データを訓練用とテスト用に分割してみましょう。

特徴量は X、目的変数は y として格納された NumPy 配列を用意してあります。

指示

100 XP
  • sklearn.model_selection から train_test_split をインポートします。
  • X と y を訓練用とテスト用に分割します。test_size は 20%、random_state は 42 に設定し、目的ラベルの比率が元のデータセットを反映するようにします。
  • knn モデルを訓練データにフィットします。
  • テストデータに対するモデルの精度を計算して表示します。