開始使用免費開始

評估分類樹

現在你已經訓練了第一棵分類樹,接下來要在測試集上評估它的表現。你會使用準確率這個指標,也就是模型在測試集中做對的預測所佔的比例。

先前練習中訓練好的模型 dt 已載入到你的工作環境,另外還有測試集的特徵矩陣 X_test 與標籤陣列 y_test

本練習屬於課程

Machine Learning with Tree-Based Models in Python

檢視課程

練習說明

  • sklearn.metrics 匯入函式 accuracy_score

  • 預測測試集的標籤,並將得到的陣列指定給 y_pred

  • 透過呼叫 accuracy_score() 評估 dt 的測試集準確率,並將結果指定給 acc

動手互動練習

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

# Import accuracy_score
from ____.____ import ____

# Predict test set labels
y_pred = ____.____(____)

# Compute test set accuracy  
acc = ____(____, ____)
print("Test set accuracy: {:.2f}".format(acc))
編輯並執行程式碼