開始使用免費開始

分開特徵與目標變數

現在你已經將資料分成訓練與測試集,接下來在擬合模型前的最後一步,是把特徵與目標變數分成不同的資料集。你會使用已為你載入的欄位名稱清單。

主要資料集已載入為 telcom,並分割為訓練與測試資料集,分別以 pandas 的 DataFrame 形式載入為 traintesttargetcustid 清單分別包含目標變數名稱與顧客 ID。你需要建立 cols 清單,放入其餘欄位的名稱。可以在主控台中自由探索這些資料集。

本練習屬於課程

Python 的行銷機器學習

檢視課程

練習說明

  • telcom 的欄位名稱存成清單,排除目標變數與顧客 ID 的名稱。
  • 提取訓練集的特徵與目標。
  • 提取測試集的特徵與目標。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Store column names from `telcom` excluding target variable and customer ID
cols = [col for col in ___.columns if col not in ___ + target]

# Extract training features
train_X = train[___]

# Extract training target
train_Y = train[___]

# Extract testing features
test_X = test[___]

# Extract testing target
test_Y = test[___]
編輯並執行程式碼