開始使用免費開始

視覺化回歸模型效能

你已經看過如何直接評估多個模型,現在要建立三個回歸模型來預測歌曲的 "energy" 強度。

music_df 資料集已經為 "genre" 加入了虛擬變數(dummy variables)。另外,特徵與目標陣列也已建立,並已分割為 X_trainX_testy_trainy_test

下列物件已為你匯入:LinearRegressionRidgeLassocross_val_scoreKFold

本練習屬於課程

使用 scikit-learn 進行監督式學習

檢視課程

練習說明

  • 撰寫一個 for 迴圈,使用 model 作為迭代變數,model.values() 作為可迭代物件。
  • 使用該模型對訓練特徵與訓練目標陣列進行交叉驗證,將 cv 設為 KFold 物件。
  • 將模型的交叉驗證分數加入 results 清單。
  • 建立一個盒狀圖來顯示結果,x 軸標籤使用各模型的名稱。

動手互動練習

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

models = {"Linear Regression": LinearRegression(), "Ridge": Ridge(alpha=0.1), "Lasso": Lasso(alpha=0.1)}
results = []

# Loop through the models' values
for ____ in models.values():
  kf = KFold(n_splits=6, random_state=42, shuffle=True)
  
  # Perform cross-validation
  cv_scores = ____(____, ____, ____, cv=____)
  
  # Append the results
  ____.____(____)

# Create a box plot of the results
plt.____(____, labels=____.____())
plt.show()
編輯並執行程式碼